I use
tf.py_func in my networks, and it works fine in training and test mode. Then I run in my networks, and it works ftf.py_func in my networks, and it works fine in training and test mode. Then I run in my networks, and it works f