
pytorch
逼疯代码
这个作者很懒,什么都没留下…
展开
-
CenterNet训练报错KeyError: ‘Traceback (most recent call last):\n File “/home/em/anaconda3/lib/python3.5
错误描述:KeyError: 'Traceback (most recent call last):\n File "/home/em/anaconda3/lib/python3.5/site-packages/torch/utils/data/dataloader.py", line 138, in _worker_loop\n samples = collate_fn([dataset[i] for i in batch_indices])\n File "/home/em/anacond...原创 2020-07-14 14:45:53 · 5263 阅读 · 0 评论 -
pytorch加载预训练模型报错-解决
pytorch加载预训练模型时报错:File "train_refinedet.py", line 115, in train refinedet_net.vgg.load_state_dict(vgg_weights) File "/home/em/anaconda3/lib/python3.5/site-packages/torch/nn/modules/module.py", line 769, in load_state_dict self.__class__.__name_...原创 2020-05-30 11:03:44 · 2625 阅读 · 0 评论