How to solve this problem KeyError: 'conv1.weight'?

shiyinzhang / Inside-Outside-Guidance

Interactive Object Segmentation with Inside-Outside Guidance

217 stars 34 forks source link

How to solve this problem KeyError: 'conv1.weight'? #20

Open lulianLiu opened 1 year ago

lulianLiu commented 1 year ago

Traceback (most recent call last): File "train.py", line 62, in pretrained=True) File "/data/networks/mainnetwork.py", line 103, in Network conv1_weight_new[:,:3,:,:]=pretrain_dict['conv1.weight'].cpu().data KeyError: 'conv1.weight'