![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
框架比较(Cell)
文章平均质量分 74
liutt233
小小白到老小白
展开
-
mindspore和pytorch的比较 -- 构建网络(一)
比较mindspore和torch网络建立相关的内容一、网络构建:分别使用mindspore和torch构建两个相同的网络:mindspore:建立一个子Cell:ConvBNReLU:from mindspore import Tensor, ops, Parameter, nnclass ConvBNReLU(nn.Cell): def __init__(self): super(ConvBNReLU, self).__init__() self.conv = nn.Co原创 2021-08-21 15:36:46 · 3270 阅读 · 0 评论 -
mindspore和pytorch的比较 -- 网络接口部分(二)
二、网络接口及支持的方法功能mindspore Cell接口torch Module接口查看模块cells_and_names()modules(),named_modules()功能一致cells()children()略有差异name_cells()name_children()略有差异获取参数parameters_and_names(),get_parameters()state_dict(destination=None)略有差异原创 2021-08-21 15:32:34 · 1221 阅读 · 1 评论