ppuda/deepnets1m/net.py [130:138]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            nn.ReLU(inplace=True)
        )
        self.classifier = nn.Linear(768, num_classes)

    def forward(self, x):
        assert self.training, 'this module is assumed to be used only for training'
        x = self.features(x)
        x = self.classifier(x.reshape(x.size(0), -1))
        return x
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



ppuda/deepnets1m/net.py [156:164]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            nn.ReLU(inplace=True)
        )
        self.classifier = nn.Linear(768, num_classes)

    def forward(self, x):
        assert self.training, 'this module is assumed to be used only for training'
        x = self.features(x)
        x = self.classifier(x.reshape(x.size(0), -1))
        return x
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -