torch.nn.MSELoss()函数解读
摘要:
函数作用torch.nn.MSELoss() 求predict和target之间的loss。 代码示例单个求其loss: crit = nn.MSELoss() # target = torch.Tensor(1) # target[0] = 10 # res = torch.Tensor(1) # 阅读全文
posted @ 2020-10-30 15:22 cltt 阅读(8018) 评论(0) 推荐(0) 编辑