Gostaria de saber como salvar a rede neural e seu estado apos os testes para que eu possa utiliza-la fora desse arquivo original
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'))