Spaces:

gagan3012
/

summarization

Runtime error

gagan3012 commited on Aug 17, 2021

Commit

037f049

1 Parent(s): 2c04c2f

Update train_model.py

Files changed (1) hide show

src/models/train_model.py CHANGED Viewed

@@ -23,6 +23,8 @@ def train_model():
         model_type=params["model_type"], model_name=params["model_name"]
     )
     model.train(
         train_df=train_df,
         eval_df=eval_df,

         model_type=params["model_type"], model_name=params["model_name"]
     )
+    print(train_df.shape, eval_df.shape)
     model.train(
         train_df=train_df,
         eval_df=eval_df,