Skip to content

Commit a52581a

Browse files
committed
disable eager execution
Signed-off-by: Arunabh <[email protected]>
1 parent da8d9d9 commit a52581a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/mnist_transferability.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@
1212
import numpy as np
1313
import tensorflow as tf
1414

15+
tf.compat.v1.disable_eager_execution()
16+
1517
from art.attacks.evasion import DeepFool
1618
from art.estimators.classification import KerasClassifier, TensorFlowClassifier
1719
from art.utils import load_mnist
@@ -60,7 +62,7 @@ def cnn_mnist_k(input_shape):
6062

6163

6264
# Get session
63-
session = tf.Session()
65+
session = tf.compat.v1.Session()
6466
k.set_session(session)
6567

6668
# Read MNIST dataset

0 commit comments

Comments
 (0)