From 63c64ce64419f21c37c3e80168963ceaaedad0d2 Mon Sep 17 00:00:00 2001 From: Sejik <52148895+Sejik@users.noreply.github.com> Date: Tue, 10 Sep 2019 18:25:48 +0900 Subject: [PATCH 1/2] Update experiment.py typo trainiing -> training typo """ trainiing --> training """ --- claf/learn/experiment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/claf/learn/experiment.py b/claf/learn/experiment.py index 06203fe..f4f2a97 100644 --- a/claf/learn/experiment.py +++ b/claf/learn/experiment.py @@ -318,7 +318,7 @@ 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") + 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={}): From dd573c51f3f8d2885e78d16ce1879bb22ea3c0cf Mon Sep 17 00:00:00 2001 From: DongjunLee Date: Tue, 10 Sep 2019 21:16:01 +0900 Subject: [PATCH 2/2] Add TODO annotation for distributed training and half precision --- claf/learn/experiment.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/claf/learn/experiment.py b/claf/learn/experiment.py index f4f2a97..dac960b 100644 --- a/claf/learn/experiment.py +++ b/claf/learn/experiment.py @@ -318,6 +318,8 @@ def _set_gpu_env(self, model): num_gpu_state = num_gpu if num_gpu > 1: num_gpu_state += " (Multi-GPU)" + + # 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