diff --git a/linen_examples/wmt/main.py b/linen_examples/wmt/main.py index 59bb068e9e..9270d08a78 100644 --- a/linen_examples/wmt/main.py +++ b/linen_examples/wmt/main.py @@ -31,7 +31,7 @@ FLAGS = flags.FLAGS config_flags.DEFINE_config_file( - "config", "configs/default.py", "Training configuration.", lock_config=True) + "config", None, "Training configuration.", lock_config=True) flags.DEFINE_string("workdir", None, "Work unit directory.") flags.DEFINE_string("jax_backend_target", None, "JAX backend target to use. Can be used with UPTC.")