warning about np1.8

tensorpack · Aug 7, 2016 · ea09302 · ea09302
1 parent 9e59832
commit ea09302
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/examples/cifar-convnet.py b/examples/cifar-convnet.py
@@ -2,7 +2,6 @@
 # -*- coding: UTF-8 -*-
 # File: cifar-convnet.py
 # Author: Yuxin Wu <ppwwyyxx@gmail.com>
-import numpy
 import tensorflow as tf
 import argparse
 import numpy as np
@@ -101,7 +100,7 @@ def get_data(train_or_test, cifar_classnum):
     ds = AugmentImageComponent(ds, augmentors)
     ds = BatchData(ds, 128, remainder=not isTrain)
     if isTrain:
-        ds = PrefetchDataZMQ(ds, 5)
+        ds = PrefetchData(ds, 3, 2)
     return ds
 
 def get_config(cifar_classnum):

diff --git a/scripts/checkpoint-manipulate.py b/scripts/checkpoint-manipulate.py
@@ -4,6 +4,7 @@
 # Author: Yuxin Wu <ppwwyyxxc@gmail.com>
 
 
+import numpy as np
 from tensorpack.tfutils.varmanip import dump_chkpt_vars
 import tensorflow as tf
 import sys

diff --git a/tensorpack/RL/envbase.py b/tensorpack/RL/envbase.py
@@ -60,6 +60,7 @@ def play_one_episode(self, func, stat='score'):
             s = self.current_state()
             act = func(s)
             r, isOver = self.action(act)
+            #print r
             if isOver:
                 s = [self.stats[k] for k in stat]
                 self.reset_stat()

diff --git a/tensorpack/__init__.py b/tensorpack/__init__.py
@@ -19,3 +19,6 @@
 from .callbacks import *
 from .dataflow import *
 from .predict import *
+
+if int(numpy.__version__.split('.')[1]) < 9:
+    logger.warn("Numpy < 1.9 could be extremely slow on some tasks.")