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

removed use_float16 option

Browse source 
fix multigpu training
This commit is contained in:
Colombo 2020-01-30 07:35:33 +04:00
parent 5fe5fa131c
commit 9fd49ee3f0
3 changed files with 36 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_Quick96/Model.py
View file

@ -163,7 +163,7 @@ class QModel(ModelBase):
masked_training = True
models_opt_on_gpu = len(devices) >= 1 and all([dev.total_mem_gb >= 2 for dev in devices])
models_opt_on_gpu = len(devices) >= 1 and all([dev.total_mem_gb >= 4 for dev in devices])
models_opt_device = '/GPU:0' if models_opt_on_gpu and self.is_training else '/CPU:0'
optimizer_vars_on_cpu = models_opt_device=='/CPU:0'