2021年6月10日
resnet mindspore
摘要： resnet50 pytorch版本实现 torchvision.models.resnet import torch.nn as nn import math import torch.utils.model_zoo as model_zoo __all__ = ['ResNet', 'resne 阅读全文
posted @ 2021-06-10 12:33 清心lh 阅读(203) 评论(0) 推荐(1) 编辑
nn.CrossEntropyLoss()->nn.SoftmaxCrossEntropyWithLogits
摘要： 问题描述： 在pytorch中，利用nn.CrossEntropyLoss()求取的loss值是一个tensor标量，但是利用mindspore中nn.SoftmaxCrossEntropyWithLogits()求取的loss值是一个矢量（因为batct_size是32，所以求得的loss也是一个 阅读全文
posted @ 2021-06-10 12:23 清心lh 阅读(390) 评论(0) 推荐(0) 编辑