大佬,运行你的ssdpytorch代码在train.py训练时报错
错误提示:
Traceback (most recent call last):
File "D:/python_code/ssd网络/ssd-pytorch-master/train.py", line 111, in
Process finished with exit code 1 附batch代码 for iteration, batch in enumerate(gen): if Use_Data_Loader: train_dataset = SSDDataset(lines[:num_train], (Config["min_dim"], Config["min_dim"])) gen = DataLoader(train_dataset, batch_size=Batch_size, num_workers=8, pin_memory=True, drop_last=True, collate_fn=ssd_dataset_collate)
我试了一下没错啊……你这个是SSD的代码么?
是的,是在你的github上面下载的,pytorch版本的ssd模型
可是我试了一下没这个错,你再下一次看看……
对了,大佬,我CUDA选择的是flose,这个不影响吧
cuda 选择的是false,说错了
不影响
重新解压,运行了一遍就可以了,谢谢大佬