test

events.out.tfevents.1667934713.t1v-n-98890786-w-0.1515223.0.v2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3be7414f5d7c27bbf6cb1a7f2c46ddb7ce0631e60a6a60eed140fd0ab4103ed0
+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