Fiz uma rede que trabalha com NLP, mas não sei com salvar os pesos depois de ter otimizado a rede.
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