前传判断:
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模式
阅读 54
2022-04-25
前传判断:
def forward(self, x): # [3,112,112]
if self.training:
print("train")
else:
print("eval")
调用:
if __name__ == "__main__":
net = MFN_85m()
net.eval()
相关推荐
精彩评论(0)