Fiz uma rede que trabalha com NLP, mas não sei com salvar os pesos depois de ter otimizado a rede.
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!
Solucionado
Fiz uma rede que trabalha com NLP, mas não sei com salvar os pesos depois de ter otimizado a rede.
Para salvar: torch.save(model.state_dict(), PATH)
Para utilizar o arquivo salvo:
model.load_state_dict(torch.load(PATH))
model.eval()
(Obs: o arquivo a ser salvo é .pt)
Esta explicadinho aqui https://pytorch.org/tutorials/beginner/saving_loading_models.html