hbaudhuin / lamboisenet Goto Github PK

Traceback (most recent call last):
File "predict.py", line 84, in
predict(model=model,
File "predict.py", line 29, in predict
mask_predicted = model(image)
File "/home/orkhan/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/orkhan/Desktop/LamboiseNet-master/Models/lightUnetPlusPlus.py", line 58, in forward
x0_3 = self.layer0_3(self.multiple_cat([x0_0, x0_2, self.upsampling(x1_2)]))
File "/home/orkhan/Desktop/LamboiseNet-master/Models/lightUnetPlusPlus.py", line 74, in multiple_cat
to_be_cat = self.crop_and_cat(to_be_cat, array[i])
File "/home/orkhan/Desktop/LamboiseNet-master/Models/lightUnetPlusPlus.py", line 86, in crop_and_cat
x1 = F.pad(x1, [diffX // 2, diffX - diffX // 2,
File "/home/orkhan/.local/lib/python3.8/site-packages/torch/nn/functional.py", line 3552, in _pad
return _VF.constant_pad_nd(input, pad, value)
RuntimeError: CUDA out of memory. Tried to allocate 208.00 MiB (GPU 0; 1.96 GiB total capacity; 833.86 MiB already allocated; 140.94 MiB free; 1.01 GiB reserved in total by PyTorch)

(python3.6) root@0b492a6ce75b:/work/LamboiseNet-master# python predict.py
Traceback (most recent call last):
File "predict.py", line 9, in
from Models.lightUnetPlusPlus import lightUnetPlusPlus
File "/work/LamboiseNet-master/Models/lightUnetPlusPlus.py", line 4, in
from Models.basicUnet import Downscaling_layer, DoubleConvolutionLayer, FinalLayer
File "/work/LamboiseNet-master/Models/basicUnet.py", line 1, in
from torch import *
AttributeError: module 'torch' has no attribute 'unique_dim'

why 'from torch import *'?

I am getting this error when running the command python train.py

last_truths[i] = ground_truth # Keep track to save the masks as images
IndexError: list assignment index out of range

Hello,could you share this paper?Thanks