bubbliiiing / efficientdet-pytorch

这是一个efficientdet-pytorch的源码,可以用于训练自己的模型。
MIT License
305 stars 61 forks source link

运行train.py出现问题 #36

Open xiangyonghong123 opened 2 years ago

xiangyonghong123 commented 2 years ago

runfile('E:/A/efficientdet-pytorch-master/train.py', wdir='E:/A/efficientdet-pytorch-master') Reloaded modules: nets, nets.efficientdet, utils, utils.anchors, nets.efficientnet, nets.layers, nets.efficientdet_training Traceback (most recent call last):

File "D:\software\Anaconda\envs\torch1.2\lib\site-packages\torch\utils\tensorboard__init__.py", line 2, in from tensorboard.summary.writer.record_writer import RecordWriter # noqa F401

ModuleNotFoundError: No module named 'tensorboard.summary'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):

File "E:\A\efficientdet-pytorch-master\train.py", line 16, in from utils.callbacks import LossHistory

File "E:\A\efficientdet-pytorch-master\utils\callbacks.py", line 9, in from torch.utils.tensorboard import SummaryWriter

File "D:\software\Anaconda\envs\torch1.2\lib\site-packages\torch\utils\tensorboard__init__.py", line 4, in raise ImportError('TensorBoard logging requires TensorBoard with Python summary writer installed. '

ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed. This should be available in 1.14 or above. image

xiangyonghong123 commented 2 years ago

并且在下载efficientdet-d0.pth后,运行predict.py出现以下问题: image

xiangyonghong123 commented 2 years ago

我重新安装环境并使用vscode后可以进行train.py的运行,训练一轮后出现以下错误: image

我将对应环境的libiomp5md.dll删除后,可以进行接下来的训练,但是每一次的训练都会出现停止再训练,请问该怎么去更正呢? image image

bubbliiiing commented 2 years ago

你的环境是什么?

xiangyonghong123 commented 2 years ago

你的环境是什么? image 按照您的博客进行的配置

xiangyonghong123 commented 2 years ago

你的环境是什么?

《深度学习环境配置2——windows下的torch=1.2.0环境配置》

bubbliiiing commented 2 years ago

显卡?

bubbliiiing commented 2 years ago

cuda和cudnn,根据问题汇总自查一下是什么