running main_ce.py does not work

[gr986]

(pytorch) D:\g\project\333>python SupContrast\main_ce.py
Files already downloaded and verified
Traceback (most recent call last):
File "SupContrast\main_ce.py", line 347, in
main()
File "SupContrast\main_ce.py", line 316, in main
loss, train_acc = train(train_loader, model, criterion, optimizer, epoch, opt)
File "SupContrast\main_ce.py", line 223, in train
output = model(images)
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 550, in call
result = self.forward(*input, **kwargs)
File "D:\g\project\333\SupContrast\resnet_big.py", line 198, in forward
return self.fc(self.encoder(x))
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 550, in call
result = self.forward(*input, **kwargs)
File "D:\g\project\333\SupContrast\resnet_big.py", line 118, in forward
out = self.layer1(out)
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 550, in call
result = self.forward(*input, **kwargs)
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\container.py", line 100, in forward
input = module(input)
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 550, in call
result = self.forward(*input, **kwargs)
File "D:\g\project\333\SupContrast\resnet_big.py", line 66, in forward
out += self.shortcut(x)
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 550, in call
result = self.forward(*input, **kwargs)
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\container.py", line 100, in forward
input = module(input)
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 550, in call
result = self.forward(*input, **kwargs)
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\conv.py", line 353, in forward
return self._conv_forward(input, self.weight)
File "C:\Users\11708\Desktop\minikangda\envs\pytorch\lib\site-packages\torch\nn\modules\conv.py", line 350, in _conv_forward
self.padding, self.dilation, self.groups)
RuntimeError: CUDA out of memory. Tried to allocate 256.00 MiB (GPU 0; 2.00 GiB total capacity; 1.15 GiB already allocated; 21.82 MiB free; 1.26 GiB reserved in total by PyTorch)

[HobbitLong]

Basically your GPU memory is not enough, try to use GPU with larger memory, or lower the batch size (together with reducing the learning rate linearly with batch size)

[gr986]

Thanks~

…

------------------ 原始邮件 ------------------ 发件人: "HobbitLong/SupContrast" ***@***.***>; 发送时间: 2021年3月23日(星期二) 下午4:05 ***@***.***>; ***@***.******@***.***>; 主题: Re: [HobbitLong/SupContrast] running main_ce.py does not work (#71) Basically your GPU memory is not enough, try to use GPU with larger memory, or lower the batch size (together with reducing the learning rate linearly with batch size) — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.