xb = torch.flatten(xb,1). xb = self.classifier(xb). return xb. but features, avgpool and classifier are "variables" of network, then you need to do:.
確定! 回上一頁