我训练了样式转换模型，并在1000幅图像上进行了训练。我为每1000张图像保存了模型，还保存了Transform Network的最终权重。现在，我希望将其保存为模型，以便可以在应用程序中使用它，但是没有一个搜索为我提供了明确的答案。
VG1 = vgg.VGG16("/kaggle/working/transformer_weight.pth")
example = torch.rand(1, 3, 800, 800)
traced_script_module = torch.jit.script(VG1, example)
traced_script_module.save('kaggle/working')
但它给出了
RuntimeError:
Module 'Sequential' has no attribute '_modules' :
File "/kaggle/working/vgg.py", line 43
layers = {'3': 'relu1_2', '8': 'relu2_2', '15': 'relu3_3', '22': 'relu4_3'}
features = {}
for name, layer in self.features._modules.items():
~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
x = layer(x)
if name in layers:
我是初学者，已经尝试了好几天。请告诉您是否需要更多信息。 我想保存模型，以便可以将其用于android studio制作应用。