github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-11 07:37:03 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update main.py

Browse source
This commit is contained in:
iperov 2018-11-21 16:29:48 +04:00 • committed by GitHub
parent ef7fdd49a0
commit 2576a411a5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
main.py
View file

@ -68,7 +68,7 @@ if __name__ == "__main__":
arguments.target_epoch = int ( os.environ['DFL_TARGET_EPOCH'] )
if 'DFL_BATCH_SIZE' in os.environ.keys():
arguments.batch_size = int ( os.environ['DFL_TARGET_EPOCH'] )
arguments.batch_size = int ( os.environ['DFL_BATCH_SIZE'] )
from mainscripts import Trainer
Trainer.main (