Update input_preprocess.py

input_preprocess.py

@@ -245,7 +245,8 @@ def make_sample(user_idx, patch, word2id, n_patches, n_masks, patch_size, gen_ra
                 input_ids[pos] = np.random.rand(patch_size)
             elif rnd_num < 0.9:
                 input_ids[pos] = word2id['[MASK]']
-    print(f'masked_pos: {masked_pos}')
+    if user_idx == 0:            
+        print(f'masked_pos: {masked_pos}')
     return [input_ids, masked_tokens, masked_pos]