Spaces:

gagan3012
/

summarization

Runtime error

gagan3012 commited on Aug 10, 2021

Commit

eb97085

1 Parent(s): b974acb

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

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

         mlflow.pytorch.autolog(log_models=False)
+        with mlflow.start_run() as run:  # noqa: F841
             trainer.fit(self.T5Model, self.data_module)
     def load_model(