KeyError: 'state_dict'

[L-ED]

sh example.sh
==> Preparing data..
==> Conv Type: BlockL1Conv
==> BN Type: LearnedBatchNorm
Traceback (most recent call last):
File "imagenet.py", line 452, in
main()
File "imagenet.py", line 279, in main
model, pr_cfg = get_model(args)
File "imagenet.py", line 240, in get_model
fc_weight = ckpt['state_dict']['classifier.0.weight']
KeyError: 'state_dict'

[zyxxmu]

Please use our provided pre-trained models and change the --pretrain_model to the corresponding path.
The pre-trained models can be downloaded at MobileNet-V1, MobileNet-V2, MobileNet-V3-Large, MobileNet-V3-Small and ResNet-50.