Spaces:

Shriti09
/

TransformerModel

Paused

Shriti09 commited on Jan 19

Commit

74105da

1 Parent(s): edba68f

Update Assign12_Model

Files changed (1) hide show

Assign12_Model.py CHANGED Viewed

@@ -271,7 +271,6 @@ if __name__ == "__main__":
                     if best_loss < target_loss:
                         torch.save(model.state_dict(), "trained_model.pth")
                         print(f"Target loss reached: {best_loss:.4f}. Model saved.")
-                        break
     print( loss )

                     if best_loss < target_loss:
                         torch.save(model.state_dict(), "trained_model.pth")
                         print(f"Target loss reached: {best_loss:.4f}. Model saved.")
     print( loss )