1 files changed, 5 insertions, 0 deletions
diff --git a/train.py b/train.py
index 91fe483..bcf921c 100755
--- a/train.py
+++ b/train.py
@@ -4,6 +4,11 @@
 #
 # usage: train.py [-h] -i DATA -o MODEL [-m M]
 
+import warnings
+with warnings.catch_warnings():
+    warnings.filterwarnings("ignore", category=DeprecationWarning)
+    import imp
+
 import matplotlib.pyplot as plt
 from mpl_toolkits.mplot3d import Axes3D
 import sys