docs: clarification instructions for TF 1.15 eager mode
@@ -945,9 +945,11 @@ class TFKerasTrial(det.Trial):
Trials default to using eager execution with TensorFlow 2.x but not with
TensorFlow 1.x. To override the default behavior, call the appropriate
- function in your ``__init__`` method. For example, if you want to disable
+ function at the top of your code. For example, if you want to disable
eager execution while using TensorFlow 2.x, call
- ``tf.compat.v1.disable_eager_execution`` at the top of your ``__init__`` method.
+ ``tf.compat.v1.disable_eager_execution`` after your import statements.
+ If you are using TensorFlow 1.x in eager mode, please add
+ ``experimental_run_tf_function=False`` to your model compile function.
For more information on writing ``tf.keras`` trial classes, refer to the
:ref:`tutorial <tf-mnist-tutorial>`.
