pytorch net save

发表于 2018-08-Fri | 阅读次数:

class NetALL:
	__init__(self):
		self.net1 = xx
		self.net2 = xx

1 2	torch.save(self.net1.dict()) w1: torch.save(self.dict()) net1.w1:

请作者喝一杯咖啡☕️