CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

leras.nn : support for tf ver 1

Browse Source
This commit is contained in:
iperov
2021-01-01 17:37:12 +04:00
parent 241d1a9c35
commit 8ff34be5e4
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/leras/nn.py
+8 -4
View File
@@ -76,15 +76,19 @@ class nn():
if first_run: if first_run:
io.log_info("Caching GPU kernels...") io.log_info("Caching GPU kernels...")
#import tensorflow as tf import tensorflow
import tensorflow.compat.v1 as tf if tensorflow.VERSION[0] == '2':
tf = tensorflow.compat.v1
else:
tf = tensorflow
import logging import logging
# Disable tensorflow warnings # Disable tensorflow warnings
tf_logger = logging.getLogger('tensorflow') tf_logger = logging.getLogger('tensorflow')
tf_logger.setLevel(logging.ERROR) tf_logger.setLevel(logging.ERROR)
tf.disable_v2_behavior() if tensorflow.VERSION[0] == '2':
tf.disable_v2_behavior()
nn.tf = tf nn.tf = tf
# Initialize framework # Initialize framework