Change VGG11_2D to VGG11_3D

[tphankr]

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

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