Как я могу автоматически скопировать строки кода в pdf или txt, чтобы сохранить архитектуру модели нейронной сети? - PullRequest
Новая
       21

Как я могу автоматически скопировать строки кода в pdf или txt, чтобы сохранить архитектуру модели нейронной сети?

0 голосов
/ 29 мая 2020

Я хотел бы экспортировать несколько строк кода в формат txt или pdf.

Я знаю, что есть возможность открыть файл txt и написать или создать PDF-файл и писать изображения или графику, но у меня это не работает.

В частности, я хотел бы сохранить модель нейронной сети , чтобы позже вспомнить, как она была настроена, а затем: 

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()

        # Capa convolucional de entrada (profundidad imagen entrada, profundidad imagen salida, kernel, padding, stride) --> 800 x 800 x 3
        self.conv1 = nn.Conv2d(3, 128, 3, padding=1)  # TODO

        # Capa convolucional --> 400 x 400 x 64
        self.conv2 = nn.Conv2d(128, 256, 3,padding=1)  # TODO

        # Capa convolucional --> 200 x 200 x 128
        self.conv3 = nn.Conv2d(256, 512, 3,padding=1)  # TODO

        # Max pooling (Kernel, stride)
        self.pool = nn.MaxPool2d(2, 2)  # TODO

        # Capa fully conected
        self.fc1 = nn.Linear(512 * 16 *16,1000 )  # TODO

        # Capa fully conected salida (tamaño, numero de clases)
        self.fc2 = nn.Linear(1000, n_clases)  # TODO

        # Abandono --> Evitar overfitting
        self.dropout = nn.Dropout(0.25)  # TODO

    def forward(self, x):
        # Secuencia de capas convolucionales y max pooling
        x = self.pool(F.relu(self.conv1(x)))
        x = self.pool(F.relu(self.conv2(x)))
        x = self.pool(F.relu(self.conv3(x)))

        # Se aplana la matriz --> Vector
        x = x.view(-1, 512 * 16 *16)  # TODO

        # Capa de abandono
        x = self.dropout(x)

        # Capa oculta con Relu
        x = F.relu(self.fc1(x))

        # Capa de abandono
        x = self.dropout(x)

        # Capa oculta de salida con Relu
        x = self.fc2(x)

        return x

model=Net()

Я хочу как-то сохранить эти строки кода.

Если вы знаете какую-либо альтернативу сохранению архитектуры модели, кроме записи ее в файл, она того стоит.

...