Skip to content

wateryzephyr/keras_bn_bug_repro

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

README

  • Keras version: 2.2.4
  • Tensorflow version: tensorflow-gpu=1.13.1
  • Need nvidia-docker 2 and CUDA 10.0 compatible driver.

To repro bug:

$ cd <repo_root>

$ cd docker

$ make build

$ make run

# Inside docker
$ cd /keras_debug

$ python repro.py

# To show the slowdown:
$ python repro.py --infer

About

Debug Keras BN slowness

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published