Open hl312 opened 3 years ago
hi , question descirption : states = torch.load(args.pretrained_model)['model'] --> error line error message: RuntimeError: CUDA error: all CUDA-capable devices are busy or unavailable please give me a hand .
hi , question descirption :
states = torch.load(args.pretrained_model)['model'] --> error line error message: RuntimeError: CUDA error: all CUDA-capable devices are busy or unavailable please give me a hand .