前传判断:
def forward(self, x): # [3,112,112]
if self.training:
print("train")
else:
print("eval")
调用:
if __name__ == "__main__":
net = MFN_85m()
net.eval()
前传判断:
def forward(self, x): # [3,112,112]
if self.training:
print("train")
else:
print("eval")
调用:
if __name__ == "__main__":
net = MFN_85m()
net.eval()