mattmacy / vnet.pytorch

A PyTorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation
https://mattmacy.github.io/vnet.pytorch
BSD 3-Clause "New" or "Revised" License
688 stars 201 forks source link

model complie error #27

Open luckwyy opened 4 years ago

luckwyy commented 4 years ago

when I prepare a train data,it always show: image

image

qianqianCDQ commented 4 years ago

when I prepare a train data,it always show: image

image

I also encountered this problem today, may you solve it now?

luckwyy commented 4 years ago

Sorry, I cannot solve it...
I think may it is the python version question, may the vnet code run in python 3.6...
you can have a try.

Chuyun-Shen commented 3 years ago

Maybe just comment out the 22th line "super(ContBatchNorm3d, self)._check_input_dim(input)" in vnet.py.