From 24856ecd0a5f79800c60352aa72821ab1d623c09 Mon Sep 17 00:00:00 2001
From: DilanBoskan <boskan.dilan@gmail.com>
Date: Wed, 2 Dec 2020 17:11:22 +0100
Subject: [PATCH] Attempt at fix for #34

---
 inference_v2.py | 3 ++-
 inference_v4.py | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/inference_v2.py b/inference_v2.py
index a9d196c..7ffb851 100644
--- a/inference_v2.py
+++ b/inference_v2.py
@@ -51,7 +51,7 @@ data = {
     # Constants
     'sr': 44_100,
     'hop_length': 1_024,
-    'window_size': 512,
+    'window_size': 320,
     'n_fft': 2_048,
 }
 default_sr = data['sr']
@@ -482,4 +482,5 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
     progress_var.set(0)  # Update Progress
     text_widget.write(f'Conversion(s) Completed and Saving all Files!\n')  # nopep8 Write Command Text
     text_widget.write(f'Time Elapsed: {time.strftime("%H:%M:%S", time.gmtime(int(time.perf_counter() - stime)))}')  # nopep8
+    torch.cuda.empty_cache()
     button_widget.configure(state=tk.NORMAL)  # Enable Button
\ No newline at end of file
diff --git a/inference_v4.py b/inference_v4.py
index cfd9b10..ffe6539 100644
--- a/inference_v4.py
+++ b/inference_v4.py
@@ -171,7 +171,7 @@ data = {
     # Constants
     'sr': 44_100,
     'hop_length': 1_024,
-    'window_size': 512,
+    'window_size': 320,
     'n_fft': 2_048,
 }
 default_sr = data['sr']
@@ -518,4 +518,5 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
     progress_var.set(0)
     text_widget.write(f'Conversion(s) Completed and Saving all Files!\n')
     text_widget.write(f'Time Elapsed: {time.strftime("%H:%M:%S", time.gmtime(int(time.perf_counter() - stime)))}')  # nopep8
+    torch.cuda.empty_cache()
     button_widget.configure(state=tk.NORMAL)  # Enable Button
\ No newline at end of file