KeyError: 'net'

[rishavkumar-7]

When running:
python main.py configs/clrnet/clr_resnet101_culane.py --test --load_from [path]/resnet101-5d3b4d8f.pth --gpus 0
gives the error KeyError:'net'
Solution:
net.load_state_dict(pretrained_model['net'], strict=False) :-> remove "['net']" this part and it will work.