Joshua Jacobs, Ph.D.

Altering utils/evaluator.py gives the desired functionality (starting line 256): ` cuda_available = torch.cuda.is_available() if cuda_available: model = torch.load(coronal_model_path) torch.cuda.empty_cache() model.cuda(device) else: model = torch.load(coronal_model_path, map_location=torch.device('cpu')) `

Thanks! Digging in!