Skip to content

Este código cria uma rede neural simples com três camadas totalmente conectadas.

License

Notifications You must be signed in to change notification settings

alexjosesilva/pytorch_SimpleNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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