search

PingoLH / FCHarDNet

Fully Convolutional HarDNet for Segmentation in Pytorch
MIT License
195 stars 52 forks source link

I can operate on *.pkl files but not *.pth files ? #56

Open jfhauris opened 3 years ago

jfhauris commented 3 years ago

@PingoLH How do you run the code with the model: hardnet_petite_base.pth I get the following error:

Traceback (most recent call last):
  File "test.py", line 139, in <module>
    test(args)
  File "test.py", line 47, in test
    device, model, loader = init_model(args)
  File "test.py", line 34, in init_model
    state = convert_state_dict(torch.load(args.model_path, map_location=device)["model_state"])
KeyError: 'model_state'