github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 14:24:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

no message

Browse source
This commit is contained in:
Jeremy Hummel 2019-09-14 16:37:39 -07:00
commit a4503223dd
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

@ -135,7 +135,7 @@ if __name__ == "__main__":
device_args = {'cpu_only' : arguments.cpu_only,
'force_gpu_idx' : arguments.force_gpu_idx,
}
from mainscripts import Trainer
from mainscripts import FlaskTrainer as Trainer
Trainer.main(args, device_args)
p = subparsers.add_parser( "train", help="Trainer")