diff --git a/claf/learn/experiment.py b/claf/learn/experiment.py index 06203fe..dac960b 100644 --- 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={}):