Traceback (most recent call last):
File "/home/boyang/Beatrix/defenses/STRIP/STRIP_our.py", line 327, in <module>
main()
File "/home/boyang/Beatrix/defenses/STRIP/STRIP_our.py", line 278, in main
list_entropy_trojan, list_entropy_benign = strip(opt, mode)
File "/home/boyang/Beatrix/defenses/STRIP/STRIP_our.py", line 198, in strip
state_dict = torch.load(opt.ckpt_path)
File "/opt/anaconda3/envs/ndg/lib/python3.9/site-packages/torch/serialization.py", line 789, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File "/opt/anaconda3/envs/ndg/lib/python3.9/site-packages/torch/serialization.py", line 1131, in _load
result = unpickler.load()
File "/opt/anaconda3/envs/ndg/lib/python3.9/site-packages/torch/serialization.py", line 1101, in persistent_load
load_tensor(dtype, nbytes, key, _maybe_decode_ascii(location))
File "/opt/anaconda3/envs/ndg/lib/python3.9/site-packages/torch/serialization.py", line 1083, in load_tensor
wrap_storage=restore_location(storage, location),
File "/opt/anaconda3/envs/ndg/lib/python3.9/site-packages/torch/serialization.py", line 215, in default_restore_location
result = fn(storage, location)
File "/opt/anaconda3/envs/ndg/lib/python3.9/site-packages/torch/serialization.py", line 187, in _cuda_deserialize
return obj.cuda(device)
File "/opt/anaconda3/envs/ndg/lib/python3.9/site-packages/torch/_utils.py", line 80, in _cuda
untyped_storage = torch.UntypedStorage(
RuntimeError: CUDA error: out of memory