FLY__1997的博客

torch 笔记 一、模型操作 添加层 b = torchvision.models.alexnet().features d = torchvision.models.alexnet().avgpool for i in d.named_modules(): b.add_module('12',i[1])#12为位置索引 改变层结构 b[1] = Linear(2048, 10) 冻结卷积层参数 for i in range(10):#具体层数要看模型结构 if isintance(b[i