2022年2月23日
Pytorch一个优化器中为不同的参数设置不同的学习率
摘要： 参考博客 阅读全文
posted @ 2022-02-23 16:46 zae 阅读(53) 评论(0) 推荐(0) 编辑
pytorch中CrossEntropyLoss的使用
摘要： csdn CrossEntropyLoss 等价于 softmax+log+NLLLoss LogSoftmax等价于softmax+log # 首先定义该类 loss = torch.nn.CrossEntropyLoss() #然后传参进去 loss(input, target) input维度 阅读全文
posted @ 2022-02-23 00:15 zae 阅读(130) 评论(0) 推荐(0) 编辑