@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'
@PingoLH How do you run the code with the model: hardnet_petite_base.pth I get the following error: