I am not able to run imageclassification.py in tftrt in the latest version of the docker tensorflow-19.01-py3.
Traceback (most recent call last):
File “image_classification.py”, line 30, in
import official.resnet.imagenet_main
File “/workspace/models/official/resnet/imagenet_main.py”, line 27, in
from official.utils.flags import core as flags_core
File “/workspace/models/official/utils/flags/core.py”, line 30, in
from official.utils.flags import _base
File “/workspace/models/official/utils/flags/_base.py”, line 25, in
from official.utils.logs import hooks_helper
File “/workspace/models/official/utils/logs/hooks_helper.py”, line 29, in
from official.utils.logs import hooks
File “/workspace/models/official/utils/logs/hooks.py”, line 28, in
class ExamplesPerSecondHook(tf.estimator.SessionRunHook):
AttributeError: module ‘tensorflow.python.estimator.api.estimator’ has no attribute ‘SessionRunHook’
I am able to run the same code in 18.01-py2. What changes do i have to make in 19.01 and tf1.12.?