Update README.md

README.md

@@ -214,11 +214,11 @@ with torch.cuda.amp.autocast(dtype=torch.bfloat16):
 Tune the masking settings:
 
 ```python
-
 xeus_model.masker.mask_prob = 0.65 # default 0.8
 xeus_model.masker.mask_length = 20 # default 10
-xeus_model.masker.mask_selection = 'static' # default uniform
+xeus_model.masker.mask_selection = 'static' # default 'uniform'
 xeus_model.train()
+feats = xeus_model.encode(wavs, wav_lengths, use_mask=True, use_final_output=False)[0][-1]
 ```
 
 ## Results