versae's picture
Saving weights and logs of step 1000
8d3cac8
raw
history blame
446 Bytes
diff --git a/run_mlm_flax.py b/run_mlm_flax.py
index ff4d2b8..d225eef 100644
--- a/run_mlm_flax.py
+++ b/run_mlm_flax.py
@@ -553,7 +553,7 @@ def main():
wandb.init(
entity='versae',
project='roberta-base-ncc',
- sync_tensorboard=False,
+ sync_tensorboard=True,
)
wandb.config.update(training_args)
wandb.config.update(model_args)