Commit 4bf2d3bc authored by Ruan Silva's avatar Ruan Silva
Browse files

run black to fix #411

parent 660894a1
1 merge request!485run black to fix #411
Showing with 4 additions and 2 deletions
+4 -2
...@@ -505,7 +505,7 @@ def _run_evaluations( ...@@ -505,7 +505,7 @@ def _run_evaluations(
return return
assert eval_module is not None, "--eval-module needs to be set." assert eval_module is not None, "--eval-module needs to be set."
module = importlib.import_module(eval_module) module = importlib.import_module(eval_module)
if not hasattr(module, 'eval_fn'): if not hasattr(module, "eval_fn"):
raise RuntimeError( raise RuntimeError(
f"{eval_module} must have a function called eval_fn to utilize for evaluations. " f"{eval_module} must have a function called eval_fn to utilize for evaluations. "
"It expects the following signature:\n" "It expects the following signature:\n"
......
...@@ -509,7 +509,9 @@ class CheckpointConfig(MetaseqDataclass): ...@@ -509,7 +509,9 @@ class CheckpointConfig(MetaseqDataclass):
) )
evaluate_last_checkpoint: bool = field( evaluate_last_checkpoint: bool = field(
default=False, default=False,
metadata={"help": "run the eval_fn from eval_module at the end of the training run"}, metadata={
"help": "run the eval_fn from eval_module at the end of the training run"
},
) )
keep_last_epochs: int = field( keep_last_epochs: int = field(
default=-1, metadata={"help": "keep only the last N epoch checkpoints"} default=-1, metadata={"help": "keep only the last N epoch checkpoints"}
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment