test on cpu

[samuro95]

Hi, I am trying to test the trained model on cpu.
In load_param in net.py, I add :
checkpoint = torch.load(path, map_location=torch.device('cpu'))

But I get :

Traceback (most recent call last):
File "UDT.py", line 122, in
net.load_param(args.model)
File "/Users/samhurault/Documents/arpe/UDT_pytorch/track/net.py", line 68, in load_param
checkpoint = torch.load(path, map_location=torch.device('cpu'))
File "/Users/samhurault/miniconda3/lib/python3.7/site-packages/torch/serialization.py", line 426, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "/Users/samhurault/miniconda3/lib/python3.7/site-packages/torch/serialization.py", line 613, in _load
result = unpickler.load()
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xc3 in position 4: invalid continuation byte