Gostaria de saber como salvar a rede neural e seu estado apos os testes para que eu possa utiliza-la fora desse arquivo original
Você está vendo a versão anterior da nova experiência da Alura que estamos preparando para você. Em breve, ela ganha uma identidade visual novinha totalmente pensada em potencializar seus estudos!
Gostaria de saber como salvar a rede neural e seu estado apos os testes para que eu possa utiliza-la fora desse arquivo original
Aqui tem um tutorial explicando essa parte: https://pytorch.org/tutorials/beginner/saving_loading_models.html
Mas essencialmente você vai salvar o state_dict() do seu modelo usando a função torch.save(). Dessa forma:
torch.save(model.state_dict(), 'caminho/pra/salvar/modelo.pt')
Na hora de carregar é só usar as funções load_state_dict() junto com o torch.load(). Fica assim:
model.load_state_dict(torch.load('caminho/pra/salvar/modelo.pt'))