hello, when I python full-supervised.py, the error message occurs:
cuda:0 pretrained/673e5d3d5b814bfca489d053f0f3a92f.pt
Traceback (most recent call last):
File "full-supervised.py", line 117, in
acc_list.append(train(datastr,splitstr))
File "full-supervised.py", line 73, in train
features = features.to(device)
RuntimeError: CUDA error: out of memory
my cuda10.0 torch1.3.1+cu100,python3.6,
do you know how to solve it?
hello, when I python full-supervised.py, the error message occurs: cuda:0 pretrained/673e5d3d5b814bfca489d053f0f3a92f.pt Traceback (most recent call last): File "full-supervised.py", line 117, in
acc_list.append(train(datastr,splitstr))
File "full-supervised.py", line 73, in train
features = features.to(device)
RuntimeError: CUDA error: out of memory
my cuda10.0 torch1.3.1+cu100,python3.6, do you know how to solve it?