How to train a neural network model?

Question

Neural networks

Hello.
Here I am writing a neural network with the PyTorch library.
I figured out how to create a model, and even how to pass a signal through layers :)
Here is the class code

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)

But here's the problem, I can't find a simple learning solution :\
Function type torch.train() , or self.layer.train() (or rather, they are, but this is how I understood the stubs).

I want to find such a simple function.
Thank you!

Answer 1

Hey!
Here is a good example of how the neural network is trained:
https://pytorch.org/tutorials/beginner/blitz/cifar...

Answer 2

Recommended:
https://pytorch.org/assets/deep-learning/Deep-Lear...