Fix output of pytorch NN

microsoft · Jan 21, 2022 · 9ab4f33 · 9ab4f33
1 parent 9fd6ba9
commit 9ab4f33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qlib/contrib/model/pytorch_nn.py b/qlib/contrib/model/pytorch_nn.py
@@ -240,7 +240,7 @@ def fit(
                 R.log_metrics(val_loss=loss_val.val, step=step)
                 if verbose:
                     self.logger.info(
-                        "[Epoch {}]: train_loss {:.6f}, valid_loss {:.6f}".format(step, train_loss, loss_val.val)
+                        "[Step {}]: train_loss {:.6f}, valid_loss {:.6f}".format(step, train_loss, loss_val.val)
                     )
                 evals_result["train"].append(train_loss)
                 evals_result["valid"].append(loss_val.val)