always-prog
@always-prog
Я программист. Python, JavaScript, Django.

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

Привет.
Вот я пишу нейросеть с библиотекой 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() (точнее они есть но это как я понял заглушки).

Хочу найти такую простую функцию.
Спасибо!
  • Вопрос задан
  • 136 просмотров
Решения вопроса 1
always-prog
@always-prog Автор вопроса
Я программист. Python, JavaScript, Django.
Привет!
Вот тут хорошо показано как тренируется нейросеть:
https://pytorch.org/tutorials/beginner/blitz/cifar...
Ответ написан
Пригласить эксперта
Ответы на вопрос 1
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы
21 окт. 2020, в 22:40
300 руб./за проект
21 окт. 2020, в 22:34
25000 руб./за проект
21 окт. 2020, в 21:24
7000 руб./за проект