add utility function to save models

utils.py

@@ -240,6 +240,13 @@ def preprocess_specs(source_root, destination_root, frame_length=256, hop=1.0):
                 np.save(destination, framed_spec)
 
 
+def save(model, path):
+    try:
+        torch.save(model.module.state_dict(), path)
+    except AttributeError:
+        torch.save(model.state_dict(), path)
+
+
 if __name__=='__main__':
     # TESTS