test

events.out.tfevents.1667932351.t1v-n-98890786-w-1.1781335.0.v2

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ee2a2884488898757efca40de1cb207cf04f9195b44731a735abe0d026c0f1ca
-size 40

run_mlm_flax_stream.py

@@ -450,7 +450,11 @@ if __name__ == "__main__":
     # Store some constant
     num_epochs = int(training_args.num_train_epochs)
     train_batch_size = int(training_args.per_device_train_batch_size) * jax.device_count() * jax.process_count()
+<<<<<<< HEAD
     eval_batch_size = int(training_args.per_device_eval_batch_size) * jax.device_count() * jax_process_count()
+=======
+    eval_batch_size = int(training_args.per_device_eval_batch_size) * jax.device_count() * jax.process_count()
+>>>>>>> 3e5f60917796ec70bad7ad043cfcb5559f582cd1
 
     # define number steps per stream epoch
     num_train_steps = data_args.num_train_steps