世界怎么会有pytorch是啥这么好用的东西 好用

我们知道Keras有一个非常有好的功能是summary,可以打印显示网络结构和参数一目了然。但是pytorch是啥本身好像不支持这一点。不过幸好有一个工具叫torchsummary,可以实现和Keras几乎一样的效果

然后我们定义好网络结构之后,就可以用summary来打印显示了假设我们定义的网络结构是一个叫Generator的类。

之后就可以打印网络结构了。┅个示例结构如下:


}

今天和大家分享Reddit上的一个热帖樓主用pytorch是啥实现了基本的RL算法,而且每个算法都在一个文件夹中完成即使没有GPU,每个算法也可以在30秒内完成训练

每个算法都在一个文件中完成。

每个算法的长度可达100~150行代码

即使没有GPU,每个算法也可以在30秒内完成训练

声明:本文内容及配图由入驻作者撰写或者入驻合莋网站授权转载。文章观点仅代表作者本人不代表电子发烧友网立场。文章及其配图仅供工程师学习之用如有内容图片侵权或者其他問题,请联系本站作侵删 

}

我要回帖

更多关于 pytorch是啥 的文章

更多推荐

版权声明:文章内容来源于网络,版权归原作者所有,如有侵权请点击这里与我们联系,我们将及时删除。

点击添加站长微信