removed use_float16 option

fix multigpu training
2020-01-30 07:35:33 +04:00
parent 5fe5fa131c
commit 9fd49ee3f0
3 changed files with 36 additions and 37 deletions
@@ -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'