Jenthe commited on
Commit
070a788
1 Parent(s): dcb24e0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
README.md CHANGED
@@ -74,6 +74,12 @@ audio, sr = torchaudio.load('sample.wav') # sample rate of 16 kHz expected
74
  embedding = ecapa2(audio)
75
  ```
76
 
 
 
 
 
 
 
77
  There is no need for `ecapa2_model.eval()` or `torch.no_grad()`, this is done automatically.
78
 
79
  ### Hierarchical Feature Extraction
 
74
  embedding = ecapa2(audio)
75
  ```
76
 
77
+ The initial calls to the JIT-model can in some cases take a very long time because of optimization attempts of the compiler. If you have issues, the JIT-optimizer can be disabled as following:
78
+ ```python
79
+ with torch.jit.optimized_execution(False):
80
+ embedding = ecapa2(audio)
81
+ ```
82
+
83
  There is no need for `ecapa2_model.eval()` or `torch.no_grad()`, this is done automatically.
84
 
85
  ### Hierarchical Feature Extraction