diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py
index e1f9f3a..a8f1caa 100644
--- a/mainscripts/Trainer.py
+++ b/mainscripts/Trainer.py
@@ -1,4 +1,5 @@
 import os
+import pickle 
 import sys
 import traceback
 import queue