bump to 0.9.0.1

examples/FasterRCNN/model_frcnn.py

@@ -5,6 +5,7 @@ import tensorflow as tf
 
 from tensorpack.tfutils.summary import add_moving_summary
 from tensorpack.tfutils.argscope import argscope
+from tensorpack.tfutils.common import get_tf_version_tuple
 from tensorpack.tfutils.scope_utils import under_name_scope
 from tensorpack.models import (
     Conv2D, FullyConnected, layer_register)
@@ -220,8 +221,10 @@ def fastrcnn_predictions(boxes, scores):
             default_value=False)
         return mask
 
+    # TF bug in version 1.11, 1.12: https://github.com/tensorflow/tensorflow/issues/22750
+    parallel = 1 if (get_tf_version_tuple() in [(1, 11), (1, 12)]) else 10
     masks = tf.map_fn(f, (scores, boxes), dtype=tf.bool,
-                      parallel_iterations=10)     # #cat x N
+                      parallel_iterations=parallel)     # #cat x N
     selected_indices = tf.where(masks)  # #selection x 2, each is (cat_id, box_id)
     scores = tf.boolean_mask(scores, masks)
 

examples/FasterRCNN/train.py

@@ -496,9 +496,6 @@ if __name__ == '__main__':
     if get_tf_version_tuple() < (1, 6):
         # https://github.com/tensorflow/tensorflow/issues/14657
         logger.warn("TF<1.6 has a bug which may lead to crash in FasterRCNN if you're unlucky.")
-    if get_tf_version_tuple() == (1, 11):
-        # https://github.com/tensorflow/tensorflow/issues/22750
-        logger.warn("TF=1.11 has a bug which leads to crash in inference.")
 
     args = parser.parse_args()
     if args.config:

tensorpack/libinfo.py

@@ -54,4 +54,4 @@ except ImportError:
 
 # This line has to be the last line of the file.
 # setup.py will use it to determine the version
-__version__ = '0.9.0'
+__version__ = '0.9.0.1'