好久没做深度学习相关了, 今天参加了一个比赛。 想用tensorboard查看训练过程, 发现了几个bug,记录一下:
1. NotFoundError: Failed to create a directory: logs/fit/20190315-164851\plugins\profile\2019-03-15_16-48-53; No such file or directory这个很坑, 就是windows下不会自动建文件夹了, 这是tensorboard 2.1.0的新错误, 以前没有。 解决方法是参考了github上的issue: https://github.com/tensorflow/tensorboard/issues/2023
搬运一下:
import datetime
log_dir = os.path.join(
"logs",
"fit",
datetime.datetime.now().strftime("%Y%m%d-%H%M%S"),
)
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)
把文件夹名字通过join来生成就可以了, 迷幻的bug。
2. pkg_resources.VersionConflict: (setuptools 36.4.0 (c:\users\17210.conda\envs\bisai\lib\site-packages), Requirement.parse(‘setuptools>=41.0.0’))从命令行运行tensorboard的时候, 显示setuptools过期, 更新一下即可:
pip install --upgrade setuptools