pool5_flat = pool5.view(pool5.size(0), -1) 【1518,512,7,7】→【1518,25088】 -...

~~~jskdhg~~~

self.vgg = models.vgg16() # Remove fc8 self.vgg.classifier = nn.Sequen...