Traceback (most recent call last):


 "/home/suhao/PycharmProjects/APGAN-master/test.py", line 24, in <module>

for i, data in enumerate(dataset):


 "/home/suhao/PycharmProjects/APGAN-master/data/__init__.py", line 72, in __iter__

for i, data in enumerate(self.dataloader):


 "/home/suhao/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 336, in __next__

return self._process_next_batch(batch)


 "/home/suhao/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 357, in _process_next_batch

raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):

Traceback (most recent call last):


 "tools/train_net_step.py", line 439, in <module>

main()


 "tools/train_net_step.py", line 319, in main

load_detectron_weight(maskRCNN, args.load_detectron)


 ".../mask-rcnn.pytorch/lib/utils/detectron_weight_helper.py", line 21, in load_detectron_weight

p_tensor.copy_(torch.Tensor(src_blobs[d_name]))
RuntimeError: invalid argument 2: sizes do not match at /opt/conda/conda-bld/pytorch_1518243271935/work/torch/lib/THC/generic/THCTensorCopy.c:51
