update

gradsflow · Oct 1, 2021 · a039ad3 · a039ad3
1 parent b6bc74e
commit a039ad3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gradsflow/callbacks/progress.py b/gradsflow/callbacks/progress.py
@@ -30,7 +30,7 @@ def __init__(self, model, **kwargs):
             "[progress.percentage]{task.percentage:>3.0f}%",
             TimeRemainingColumn(),
             self.table_column,
-            refresh_per_second=kwargs.get("refresh_per_second", 5),
+            refresh_per_second=kwargs.get("refresh_per_second", 10),
             expand=kwargs.get("expand", True),
         )
         tracker.progress = self.progress
@@ -60,6 +60,7 @@ def on_train_epoch_end(self):
 
     def on_train_step_end(self):
         self.progress.update(self.train_prog_bar, advance=1)
+        self.table_column.renderable = self.model.tracker.create_table()
 
     def on_val_epoch_start(self):
         val_dl = self.model.tracker.autodataset.val_dataloader