test

events.out.tfevents.1667932958.t1v-n-98890786-w-2.1496415.0.v2

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

run_mlm_flax_stream.py

@@ -450,7 +450,7 @@ 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()
-    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()
 
     # define number steps per stream epoch
     num_train_steps = data_args.num_train_steps