warning about np1.8

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):

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

tensorpack/RL/envbase.py

@@ -60,6 +60,7 @@ class RLEnvironment(object):
             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()

tensorpack/__init__.py

@@ -19,3 +19,6 @@ from .tfutils import *
 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.")