2020年2月24日
损失函数
摘要： 1、torch.nn.CrossEntropyLoss() 用于多分类问题 loss_func=torch.nn.CrossEntropyLoss() loss=loss_func(input_data,input_target) 其中input_data的shape一般是(batch_size,o 阅读全文
posted @ 2020-02-24 16:19 6+0 阅读(283) 评论(0) 推荐(0) 编辑