tensorflow和pytorch环境是好的的话，链接中的logger.py拉到自己的工程里，train.py里添加相应代码，直接能用。
关于环境，小小折腾了下，大概一小时：
大概一年前用过tensorflow， mac里的环境还在，当时装的虚拟环境，由于工程中用到了caffe,在虚拟环境中编译caffe，去掉之前工程中用的caffe路径，把虚拟环境中的caffe路径添加到虚拟环境下的pythonpath
export PYTHONPATH=/Users/tensorflow虚拟环境中的caffe/python路径:$PYTHONPATH
pip install torch torchvision
然后进入python，依次import caffe torch tensorflow没问题就可以了