1 在torchscript 里面使用nn.modulelist时,
例如声明一个 self.vgg= nn.modulelist(....)
不允许使用如
for i in range(len(self.vgg)):
y=self.vgg[i](x)
报错:module cannot be used as a value
因为torchscript不能把module当作值对待,
应该使用
for l in self.vgg:
y = l(x)
1 在torchscript 里面使用nn.modulelist时,
例如声明一个 self.vgg= nn.modulelist(....)
不允许使用如
for i in range(len(self.vgg)):
y=self.vgg[i](x)
报错:module cannot be used as a value
因为torchscript不能把module当作值对待,
应该使用
for l in self.vgg:
y = l(x)