Small bug when running on CPU-only machines

[uzpaka]

d2-net/lib/model_test.py

Line 61 in 8198366

self.load_state_dict(torch.load(model_file)['model'])

With this line, pytorch tries to load the weights on the GPU and throws an error if it can't. The following update should solve the issue:

if model_file is not None:
if use_cuda:
self.load_state_dict(torch.load(model_file)['model'])
else:
self.load_state_dict(torch.load(model_file, map_location='cpu')['model'])

[mihaidusmanu]

This should be fixed by the latest commit. Thank you!