Spaces:

gagan3012
/

summarization

Runtime error

gagan3012 commited on Aug 10, 2021

Commit

b974acb

1 Parent(s): acc3a63

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -390,7 +390,6 @@ class Summarization:
         mlflow.pytorch.autolog(log_models=False)
-        with mlflow.start_run() as run:
             trainer.fit(self.T5Model, self.data_module)
     def load_model(

         mlflow.pytorch.autolog(log_models=False)
             trainer.fit(self.T5Model, self.data_module)
     def load_model(