search

shiba24 / 3d-unet

Chainer implementations of 3D UNet
182 stars 55 forks source link

[Bug]: Wrongly use batchnorm2d #10

Closed heqin-zhu closed 4 years ago

heqin-zhu commented 4 years ago

https://github.com/shiba24/3d-unet/blob/398dca9b1752a07a7f80c45bbca017ca0fedeed5/pytorch/model.py#L39 Should we use batchnorm3d instead of batchnorm2d in the referred line?

shiba24 commented 4 years ago

yeah i guess so. could you create a PR?

heqin-zhu commented 4 years ago

Okay! I will do it later.

shiba24 commented 4 years ago

fixed with #11