小编Kan*_*ani的帖子

我必须在使用不同设备的不同类型的 pytorch 模型上堆叠一些我自己的层。

例如，A是cuda模型，B是cpu模型（但在获得设备类型之前我不知道）。那么新模型分别是C和D，其中

class NewModule(torch.nn.Module):
    def __init__(self, base):
        super(NewModule, self).__init__()
        self.base = base
        self.extra = my_layer() # e.g. torch.nn.Linear()

    def forward(self,x):
        y = self.base(x)
        z = self.extra(y)
        return z

...

C = NewModule(A) # cuda
D = NewModule(B) # cpu

但是我必须继续前进base，并extra在相同的设备，即base和extra的C是CUDA模型和D的是CPU的型号。所以我试过这个__inin__：

def __init__(self, base):
    super(NewModule, self).__init__()
    self.base = base
    self.extra = …

我想用VS Code查看一些代码文件并编辑我自己的代码来调用它们。我不应该更改它们（包括文件模式），因此最好以只读模式打开这些文件。

其实我也因为不小心修改，遇到了一些麻烦。所以我认为我应该以更安全的方式查看这些文件。

如何用 VS 代码做到这一点？我知道 PyCharm 和 Notepad++ 已经实现了这个功能，那么 VS Code 也有吗？我什至在它的官方网站上也没有找到它。

非常感谢。