diff --git a/examples/deepswe/train_deepswe_nb.py b/examples/deepswe/train_deepswe_nb.py index 7f92333bd..8da3fc9cd 100644 --- a/examples/deepswe/train_deepswe_nb.py +++ b/examples/deepswe/train_deepswe_nb.py @@ -663,7 +663,7 @@ def transform(entry): ) metrics_logging_options = metrics_logger.MetricsLoggerOptions( - log_dir=args.metrics_logger_dir, flush_every_n_steps=2 + log_dir=args.metric_logger_dir, flush_every_n_steps=2 ) optimizer = optax.schedules.inject_hyperparams(optax.adamw)(