Slight modification. torch.laod(f = '*.pth', map_location = torch.device('cpu') added) 8ee49d1 juanpajedrez commited on May 2