pytorch比较weight

发表于 | 阅读次数: 
1
2
3
4
5
a = list(self.parameters())[0].clone()
loss.backward()
self.optimizer.step()
b = list(self.parameters())[0].clone()
torch.equal(a.data, b.data)
请作者喝一杯咖啡☕️