Loading scripts/ActorOnly+CEL/main_CrossEntropy.py +3 −0 Original line number Diff line number Diff line Loading @@ -61,6 +61,9 @@ torch.save(val_rouge_history, hist_path/'model_CE_val_rouge_hist.pt') # load best model weights m.load_state_dict(best_model_wts) #save best model torch.save(m.state_dict(), f'model_CE_wts.pth') # testing since = time.time() Loading Loading
scripts/ActorOnly+CEL/main_CrossEntropy.py +3 −0 Original line number Diff line number Diff line Loading @@ -61,6 +61,9 @@ torch.save(val_rouge_history, hist_path/'model_CE_val_rouge_hist.pt') # load best model weights m.load_state_dict(best_model_wts) #save best model torch.save(m.state_dict(), f'model_CE_wts.pth') # testing since = time.time() Loading
