Change VGG11_2D to VGG11_3D
tphankr opened this issue · 0 comments
tphankr commented
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