Pytorch-YOLOv3运行中可能遇到的错误

github链接:https://github.com/eriklindernoren/PyTorch-YOLOv3

 

错误1

File "pytorch_platform/PyTorch-YOLOv3/utils/logger.py", line 7, in __init__
    self.writer = tf.summary.FileWriter(log_dir)
AttributeError: module 'tensorboard.summary._tf.summary' has no attribute 'FileWriter'

出现上述错误的原因是我使用的tensorflow版本是2.0版本,其中一些api有改动,解决方法如下:

将    self.writer = tf.summary.FileWriter(log_dir)

改为  self.writer = tf.summary.create_file_writer(log_dir)

 

你可能感兴趣的:(深度学习算法相关)