随笔分类 - mindspore
摘要:resnet50 pytorch版本实现 torchvision.models.resnet import torch.nn as nn import math import torch.utils.model_zoo as model_zoo __all__ = ['ResNet', 'resne
阅读全文
摘要:问题描述: 在pytorch中,利用nn.CrossEntropyLoss()求取的loss值是一个tensor标量,但是利用mindspore中nn.SoftmaxCrossEntropyWithLogits()求取的loss值是一个矢量(因为batct_size是32,所以求得的loss也是一个
阅读全文

浙公网安备 33010602011771号