Skip to content

S
seminar-breakout
Commit 3f48ed30 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix avgpool issue in TF

parent d0bdf185
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tensorpack/libinfo.py
View file @ 3f48ed30
......@@ -8,5 +8,6 @@ os.environ['OPENCV_OPENCL_RUNTIME'] = ''
os.environ['TF_ENABLE_WINOGRAD_NONFUSED'] = '1' # issue#9339
os.environ['TF_AUTOTUNE_THRESHOLD'] = '3' # use more warm-up
os.environ['TF_AVGPOOL_USE_CUDNN'] = '1' # issue#8566
__version__ = '0.1.8'
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment