Edoardo-BS commited on
Commit
0f01b21
1 Parent(s): 073d423

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -48,6 +48,7 @@ hubert_ecg.load_state_dict(checkpoint['model_state_dict']) # fine-tuned model re
48
  ```
49
 
50
  ## Easier usage
 
51
  ```python
52
  from transformers import AutoModel
53
  size = 'small' # any size from small, base, large
 
48
  ```
49
 
50
  ## Easier usage
51
+ (for pre-trained models only)
52
  ```python
53
  from transformers import AutoModel
54
  size = 'small' # any size from small, base, large