PyTorch — это библиотека машинного обучения, разработанная Facebook. Она позволяет создавать нейронные сети и обучать их на графических процессорах, что ускоряет процесс обучения и улучшает точность моделей.
В этом примере мы загружаем датасет MNIST с рукописными цифрами, создаем простую нейронную сеть с двумя полносвязными слоями и функцией активации
ReLU, обучаем ее с помощью стохастического градиентного спуска, а затем тестируем результат на отложенных данных. При этом мы используем функцию потерь
CrossEntropyLoss, которая подходит для задач классификации, и оптимизатор SGD.