ssd-pytorch icon indicating copy to clipboard operation
ssd-pytorch copied to clipboard

大佬,运行你的ssdpytorch代码在train.py训练时报错

Open fzh66666 opened this issue 5 years ago • 7 comments

错误提示: Traceback (most recent call last): File "D:/python_code/ssd网络/ssd-pytorch-master/train.py", line 111, in images = batch[0] TypeError: 'ExceptionWrapper' object does not support indexing

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)

fzh66666 avatar Jul 28 '20 09:07 fzh66666

我试了一下没错啊……你这个是SSD的代码么?

bubbliiiing avatar Jul 30 '20 03:07 bubbliiiing

是的,是在你的github上面下载的,pytorch版本的ssd模型

fzh66666 avatar Jul 30 '20 03:07 fzh66666

可是我试了一下没这个错,你再下一次看看……

bubbliiiing avatar Jul 30 '20 03:07 bubbliiiing

对了,大佬,我CUDA选择的是flose,这个不影响吧

fzh66666 avatar Jul 30 '20 03:07 fzh66666

cuda 选择的是false,说错了

fzh66666 avatar Jul 30 '20 03:07 fzh66666

不影响

bubbliiiing avatar Jul 30 '20 03:07 bubbliiiing

重新解压,运行了一遍就可以了,谢谢大佬

fzh66666 avatar Jul 30 '20 03:07 fzh66666