diff --git a/src/baskerville/seqnn.py b/src/baskerville/seqnn.py
index c454672..36377fd 100644
--- a/src/baskerville/seqnn.py
+++ b/src/baskerville/seqnn.py
@@ -19,7 +19,7 @@
 from natsort import natsorted
 import numpy as np
 import tensorflow as tf
-
+import gc
 from baskerville import blocks
 from baskerville import layers
 from baskerville import metrics