-
Place source files in
files/source -
Modify
files/source/Inference.pyto make compatible with running only on CPU:
Replace on line 88
checkpoint = torch.load(model_file_name)
with
checkpoint = torch.load(model_file_name) if torch.cuda.is_available() else torch.load(model_file_name, map_location=torch.device('cpu'))