2021年6月19日
pytorch中的add_module函数
摘要： 现只讲在自定义网络中add_module的作用。 总结： 在自定义网络的时候，由于自定义变量不是Module类型（例如，我们用List封装了几个网络），所以pytorch不会自动注册网络模块。add_module函数用来为网络添加模块的，所以我们可以使用这个函数手动添加自定义的网络模块。当然，这种情 阅读全文
posted @ 2021-06-19 17:32 蒙面的普罗米修斯 阅读(10111) 评论(0) 推荐(3) 编辑