Sadjad Alikhani commited on
Commit
573514a
1 Parent(s): 78d5782

Update input_preprocess.py

Browse files
Files changed (1) hide show
  1. input_preprocess.py +2 -2
input_preprocess.py CHANGED
@@ -246,8 +246,8 @@ def make_sample(user_idx, patch, word2id, n_patches, n_masks, patch_size, gen_ra
246
  input_ids[pos] = np.random.rand(patch_size)
247
  elif rnd_num < 0.9:
248
  input_ids[pos] = word2id['[MASK]']
249
- if user_idx == 0:
250
- print(f'masked_pos: {masked_pos}')
251
  return [input_ids, masked_tokens, masked_pos]
252
 
253
 
 
246
  input_ids[pos] = np.random.rand(patch_size)
247
  elif rnd_num < 0.9:
248
  input_ids[pos] = word2id['[MASK]']
249
+ #if user_idx == 0:
250
+ # print(f'masked_pos: {masked_pos}')
251
  return [input_ids, masked_tokens, masked_pos]
252
 
253