Este código cria uma rede neural simples com três camadas totalmente conectadas. Em seguida, ele treina a rede usando o conjunto de dados MNIST para classificação de dígitos escritos à mão. Você pode expandir este exemplo adicionando mais camadas à rede, ajustando os hiperparâmetros ou usando conjuntos de dados diferentes.
Tecnologias
-
PyTorch
-
Python 3