Merge branch 'refactor/helper'

naver · Sep 10, 2019 · 4a14101 · 4a14101
2 parents e941f8f + dd573c5
commit 4a14101
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/claf/learn/experiment.py b/claf/learn/experiment.py
@@ -318,7 +318,9 @@ def _set_gpu_env(self, model):
         num_gpu_state = num_gpu
         if num_gpu > 1:
             num_gpu_state += " (Multi-GPU)"
-        logger.info(f"use_gpu: {self.config.use_gpu} num_gpu: {num_gpu_state}, distributed training: False, 16-bits trainiing: False")
+
+        # TODO: distributed training and 16-bits training (FP16)
+        logger.info(f"use_gpu: {self.config.use_gpu} num_gpu: {num_gpu_state}, distributed training: False, 16-bits training: False")
         return model
 
     def set_trainer(self, model, op_dict={}, save_params={}):