前传判断：
def forward(self, x): # [3,112,112]
if self.training:
print("train")
else:
print("eval")调用：
if __name__ == "__main__":
net = MFN_85m()
net.eval()pytorch train模式
猜你喜欢
转载自blog.csdn.net/jacke121/article/details/124411539
今日推荐
周排行