Skip to content

Instantly share code, notes, and snippets.

@Jim-Holmstroem

Jim-Holmstroem/keras_fix.py

Created March 18, 2017 21:32
Show Gist options
  • Save Jim-Holmstroem/c9f6cc1470a49503873d26165efbba1d to your computer and use it in GitHub Desktop.
Save Jim-Holmstroem/c9f6cc1470a49503873d26165efbba1d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
keras_fix.py
from keras import backend as K
K.set_image_dim_ordering('th')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment