From 9b568bc25ef1efed9b9a4f19d7ae9c1cfae69bdf Mon Sep 17 00:00:00 2001 From: yann le paih Date: Tue, 30 Nov 2021 20:12:51 +0100 Subject: [PATCH] fix regression in nn.py propose fix for error : UnboundLocalError: local variable 'tf' referenced before assignment issue describe in : https://github.com/iperov/DeepFaceLab/issues/5436 --- core/leras/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/leras/nn.py b/core/leras/nn.py index f392aaf..28f1308 100644 --- a/core/leras/nn.py +++ b/core/leras/nn.py @@ -115,7 +115,7 @@ class nn(): nn.tf_sess_config = config if nn.tf_sess is None: - nn.tf_sess = tf.Session(config=nn.tf_sess_config) + nn.tf_sess = nn.tf.Session(config=nn.tf_sess_config) if floatx == "float32": floatx = nn.tf.float32