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 | |
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) | |