(env) G:\sd_training_infor\software\BasicPBC>python inference_line_frames.py --path dataset/test/laughing_girl
dataset/test/laughing_girl\seg\0000.png created.
dataset/test/laughing_girl\seg\0001.png created.
dataset/test/laughing_girl\seg\0002.png created.
dataset/test/laughing_girl\seg\0003.png created.
Traceback (most recent call last):
File "G:\sd_training_infor\software\BasicPBC\inference_line_frames.py", line 200, in
model = BasicPBC(
File "G:\sd_training_infor\software\BasicPBC\basicsr\archs\basicpbc_arch.py", line 571, in init
state_dict = torch.load(args["ckpt"])
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 1025, in load
return _load(opened_zipfile,
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 1446, in _load
result = unpickler.load()
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 1416, in persistent_load
typed_storage = load_tensor(dtype, nbytes, key, _maybe_decode_ascii(location))
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 1390, in load_tensor
wrap_storage=restore_location(storage, location),
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 390, in default_restore_location
result = fn(storage, location)
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 265, in _cuda_deserialize
device = validate_cuda_device(location)
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 249, in validate_cuda_device
raise RuntimeError('Attempting to deserialize object on a CUDA '
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
(env) G:\sd_training_infor\software\BasicPBC>python inference_line_frames.py --path dataset/test/laughing_girl dataset/test/laughing_girl\seg\0000.png created. dataset/test/laughing_girl\seg\0001.png created. dataset/test/laughing_girl\seg\0002.png created. dataset/test/laughing_girl\seg\0003.png created. Traceback (most recent call last): File "G:\sd_training_infor\software\BasicPBC\inference_line_frames.py", line 200, in
model = BasicPBC(
File "G:\sd_training_infor\software\BasicPBC\basicsr\archs\basicpbc_arch.py", line 571, in init
state_dict = torch.load(args["ckpt"])
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 1025, in load
return _load(opened_zipfile,
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 1446, in _load
result = unpickler.load()
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 1416, in persistent_load
typed_storage = load_tensor(dtype, nbytes, key, _maybe_decode_ascii(location))
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 1390, in load_tensor
wrap_storage=restore_location(storage, location),
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 390, in default_restore_location
result = fn(storage, location)
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 265, in _cuda_deserialize
device = validate_cuda_device(location)
File "G:\sd_training_infor\software\BasicPBC\env\lib\site-packages\torch\serialization.py", line 249, in validate_cuda_device
raise RuntimeError('Attempting to deserialize object on a CUDA '
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.