Better support for virtual_batch_size

e0a7e8f9 · Yuxin Wu · 9f4600a0 · e0a7e8f9 · e0a7e8f9
Commit e0a7e8f9 authored Mar 11, 2020 by Yuxin Wu
Expand all Show whitespace changes
Inline Side-by-side

Showing with 99 additions and 76 deletions

README.md README.md +1 -1

tensorpack/models/batch_norm.py tensorpack/models/batch_norm.py +98 -75

No files found.
--- a/README.md
+++ b/README.md
@@ -67,7 +67,7 @@ Dependencies:

 + Python 3.3+.
 + Python bindings for OpenCV. (Optional, but required by a lot of features)
-+ TensorFlow ≥ 1.3, < 2. (Not required if you only want to use `tensorpack.dataflow` alone as a data processing library)
+ TensorFlow ≥ 1.5, < 2. (Not required if you only want to use `tensorpack.dataflow` alone as a data processing library)
 ```
 pip install --upgrade git+https://github.com/tensorpack/tensorpack.git
 # or add `--user` to install to user's local directories

--- a/tensorpack/models/batch_norm.py
+++ b/tensorpack/models/batch_norm.py