mattmacy / vnet.pytorch

A PyTorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation

Home Page:https://mattmacy.github.io/vnet.pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

model complie error

luckwyy opened this issue · comments

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

image

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

image

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

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.

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