Skip to content

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

StagingArea was only moved to contrib after 1.1

parent 9d0b28a0
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
tensorpack/train/input_data.py
View file @ d0bdf185
...@@ -4,7 +4,10 @@ ...@@ -4,7 +4,10 @@
# Author: Yuxin Wu <ppwwyyxxc@gmail.com> # Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from tensorflow.contrib.staging import StagingArea try:
from tensorflow.contrib.staging import StagingArea
except ImportError:
pass
from itertools import chain from itertools import chain
from abc import ABCMeta, abstractmethod from abc import ABCMeta, abstractmethod
......
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