0

我为 Style Transfer 训练了一个模型，并在 1000 张图像上对其进行了训练。我为每 1000 张图像保存了模型，还保存了变换网络的最终权重。现在我希望将它保存为模型，以便我可以在应用程序中使用它，但没有一个搜索给我一个明确的答案如何做到这一点。

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 来制作应用程序。

笔记本位于“ https://www.kaggle.com/starktony45/fast-neural-style ”

4

0 回答 0