Как тренировать модель нейросети?

Question

[Stepan Sidorov]

Привет.
Вот я пишу нейросеть с библиотекой PyTorch.
Я разобрался как создать модель, и даже как провести сигнал через слои :)
Вот код класса

class Net(nn.Module):
    def __init__(self):
        super(Net,self).__init__()
        self.layers = nn.Sequential(
            nn.Linear(in_features=1,out_features=10),
            nn.ReLU(),
            nn.Linear(in_features=10,out_features=5),
            nn.ReLU(),
            nn.Linear(in_features=5,out_features=1),
            nn.ReLU()
        )

    def forward(self, x):
        return self.layers(x)

Но вот проблема, я не могу найти простого решения обучения :\
Типа функции torch.train(), или self.layer.train() (точнее они есть но это как я понял заглушки).

Хочу найти такую простую функцию.
Спасибо!

Answer 1

[Stepan Sidorov]

Привет!
Вот тут хорошо показано как тренируется нейросеть:
https://pytorch.org/tutorials/beginner/blitz/cifar...

Answer 2

[dmshar]

Рекомендую:
https://pytorch.org/assets/deep-learning/Deep-Lear...

Comments

[Stepan Sidorov]

Спасибо!
Но там нет простой функции, вопрос был именно в ней.

[dmshar]

В книге, посвященной тому, как использовать PyTourch нет информации как тренировать и использовать эту библиотеку? Вы сами верите в то, что вы написали? Или оно просто не так просто, как хотелось бы?

[Stepan Sidorov]

dmshar, Да. Там нет простой функции.

[dmshar]

Stepan Sidorov, Нет. И как же люди им пользуются? Или вы думаете, что если в книге нет, то она есть где-то в другом месте?