Write memory usage to log

neural_acoustic_distance.py

@@ -14,6 +14,8 @@ from transformers.models.wav2vec2 import Wav2Vec2Model
 from datetime import datetime
 from random import randrange
 
+import os
+import psutil
 
 def play_audio(filename):
     audio_file = open(filename, "rb")
@@ -125,6 +127,9 @@ if model_id == "other":
                              value="facebook/wav2vec2-large-960h",
                              key="model")
 
+print(f"\n### Start new run\n") # test
+print('0. RAM memory available:', round(psutil.virtual_memory()[1] / (1024.0 ** 3), 1), 'GB') # test
+
 try:
     cfg = AutoConfig.from_pretrained(model_id)
     layer = st.number_input("Select the layer you want to use:", min_value=1, max_value=cfg.num_hidden_layers, value=10)
@@ -219,3 +224,4 @@ if filename_x is not None and filename_y is not None and layer is not None:
         btn = st.download_button(label="Download plot", data=file, file_name="plot.pdf", mime="image/pdf")
 
     print('8. End', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
+    print('9. RAM memory % used:', psutil.virtual_memory()[2]) # test