Update neural_acoustic_distance.py

neural_acoustic_distance.py

@@ -39,7 +39,7 @@ def compute_costs(gcm):
     return res, n
 
 
-@st.cache(show_spinner=False, hash_funcs={torch.nn.parameter.Parameter: lambda _: None}, max_entries=1)
+#@st.cache(show_spinner=False, hash_funcs={torch.nn.parameter.Parameter: lambda _: None}, max_entries=1)
 def load_wav2vec2_featurizer(model_id: str, layer: Optional[int] = None):
     transformers.logging.set_verbosity(transformers.logging.ERROR)
 
@@ -224,3 +224,10 @@ if filename_x is not None and filename_y is not None and layer is not None:
 
     print('8. End', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
     print(f"9. RAM used: {psutil.Process().memory_info().rss / (1024 * 1024):.2f} MB") # test
+    
+for name in dir():
+    if not name.startswith('_'):
+        del globals()[name]
+
+import gc
+gc.collect()