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

[jfhauris]

@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'