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
This commit is contained in:
yann le paih 2021-11-30 20:12:51 +01:00 committed by GitHub
commit 9b568bc25e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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