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