aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xtrain.py5
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