ternaus/TernausNet

Change VGG11_2D to VGG11_3D

tphankr opened this issue · 0 comments

I followed this code, everything good for 2D model.
Now, I want to use it for 3D model. I changed 2D to 3D, example:
self.pool = nn.MaxPool2d(2,2) => self.pool = nn.MaxPool3d(2,2)
.......
but, at the line 49 ( the picture); : models.vgg11(pretrained=pretrained).features is 2D model. I want to use vgg11 for 3D. How do I do? Thank you
image

p/s: I visualizered vgg11; it is 2D model
image