命令行输入：python train.py --config=yolact_coco_custom_config --resume=weights/yolact_plus_resnet50_54_800000.pth --only_last_layer
报错：
Traceback (most recent call last):
File "train.py", line 518, in
train()
File "train.py", line 319, in train
losses = net(datum)
File "/home/test/anaconda3/envs/yolact/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/home/test/anaconda3/envs/yolact/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 153, in forward
return self.gather(outputs, self.output_device)
File "train.py", line 175, in gather
out[k] = torch.stack([output[k].to(output_device) for output in outputs])
File "train.py", line 175, in
out[k] = torch.stack([output[k].to(output_device) for output in outputs])
KeyError: 'I'