github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-15 01:23:44 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

code refactoring,

Browse source 
lr_dropout is now disabled in pretraining mode
changed help message for lr_dropout and random_warp
This commit is contained in:
Colombo 2020-03-07 13:59:47 +04:00
parent 9ccdd271a4
commit ada60ccefe
4 changed files with 612 additions and 232 deletions
Download patch file Download diff file Expand all files Collapse all files

7
core/leras/nn.py
View file

@ -86,6 +86,7 @@ class nn():
# Models
PatchDiscriminator = None
IllumDiscriminator = None
CodeDiscriminator = None
@staticmethod
def initialize(device_config=None, floatx="float32", data_format="NHWC"):
@ -147,13 +148,15 @@ class nn():
from .initializers import initialize_initializers
from .optimizers import initialize_optimizers
from .models import initialize_models
from .archis import initialize_archis
initialize_tensor_ops(nn)
initialize_layers(nn)
initialize_initializers(nn)
initialize_optimizers(nn)
initialize_models(nn)
initialize_archis(nn)
if nn.tf_sess is None:
nn.tf_sess = tf.Session(config=nn.tf_sess_config)