Pytorch基础——从nn.module转写成.py脚本(一)
Pytorch基础——从nn.module转写成.py脚本(一)简介代码结果
简介
起因还是在做AutoML,中间有小伙伴需要改写模型中channel的数量,希望能将模型channel变成一个变量,这需要改动torch定义的model类的脚本文件,所以想能不能根据传入的model,自动生成.py的脚本。
代码
为了测试代码,我们先写一个model.py文件生成一个简单地模型。
# model.py...
原创
2020-04-02 09:56:26 ·
271 阅读 ·
0 评论