While I run this code in google colab:
!python3 test.py --model 3 --checkpoints "/content/lafin/checkpoints/example"
I got this error:
start testing...
THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=383 error=11 : invalid argument
Traceback (most recent call last):
File "test.py", line 2, in
main(mode=2)
File "/content/lafin/main.py", line 61, in main
model.test()
File "/content/lafin/src/lafin.py", line 367, in test
output_landmark = self.landmark_model(images (1 - masks) + masks, masks)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(input, kwargs)
File "/content/lafin/src/models.py", line 247, in forward
landmark_gen = self.mbnet(images_masked)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, *kwargs)
File "/content/lafin/src/networks.py", line 301, in forward
x = self.features(images) # C = 320
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(input, kwargs)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/container.py", line 92, in forward
input = module(input)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, *kwargs)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/container.py", line 92, in forward
input = module(input)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(input, **kwargs)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/conv.py", line 338, in forward
self.padding, self.dilation, self.groups)
RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:383
While I run this code in google colab: !python3 test.py --model 3 --checkpoints "/content/lafin/checkpoints/example"
I got this error: start testing...
THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=383 error=11 : invalid argument Traceback (most recent call last): File "test.py", line 2, in
main(mode=2)
File "/content/lafin/main.py", line 61, in main
model.test()
File "/content/lafin/src/lafin.py", line 367, in test
output_landmark = self.landmark_model(images (1 - masks) + masks, masks)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(input, kwargs)
File "/content/lafin/src/models.py", line 247, in forward
landmark_gen = self.mbnet(images_masked)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, *kwargs)
File "/content/lafin/src/networks.py", line 301, in forward
x = self.features(images) # C = 320
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(input, kwargs)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/container.py", line 92, in forward
input = module(input)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, *kwargs)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/container.py", line 92, in forward
input = module(input)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(input, **kwargs)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/conv.py", line 338, in forward
self.padding, self.dilation, self.groups)
RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:383
Please Help!!
Colab Specification:
[name: "/device:CPU:0" device_type: "CPU" memory_limit: 268435456 locality { } incarnation: 6642222752450203202, name: "/device:GPU:0" device_type: "GPU" memory_limit: 14674281152 locality { bus_id: 1 links { } } incarnation: 15233849231393388862 physical_device_desc: "device: 0, name: Tesla T4, pci bus id: 0000:00:04.0, compute capability: 7.5"]