我们知道Keras有一个非常有好的功能是summary,可以打印显示网络结构和参数一目了然。但是pytorch是啥本身好像不支持这一点。不过幸好有一个工具叫torchsummary,可以实现和Keras几乎一样的效果
然后我们定义好网络结构之后,就可以用summary来打印显示了假设我们定义的网络结构是一个叫Generator的类。
之后就可以打印网络结构了。┅个示例结构如下:
今天和大家分享Reddit上的一个热帖樓主用pytorch是啥实现了基本的RL算法,而且每个算法都在一个文件夹中完成即使没有GPU,每个算法也可以在30秒内完成训练
每个算法都在一个文件中完成。
每个算法的长度可达100~150行代码
即使没有GPU,每个算法也可以在30秒内完成训练
声明:本文内容及配图由入驻作者撰写或者入驻合莋网站授权转载。文章观点仅代表作者本人不代表电子发烧友网立场。文章及其配图仅供工程师学习之用如有内容图片侵权或者其他問题,请联系本站作侵删
版权声明:文章内容来源于网络,版权归原作者所有,如有侵权请点击这里与我们联系,我们将及时删除。