print sample

run_mlm_flax_stream.py

@@ -287,6 +287,7 @@ def advance_iter_and_group_samples(train_iterator, num_samples, max_seq_length):
         i += len(tokenized_samples["input_ids"])
 
         # concatenate tokenized samples to list
+        print('sample',tokenized_samples)
         samples = {
             k: samples[k] + tokenized_samples[k] for k in ["input_ids", "attention_mask", "special_tokens_mask"]
         }