Update training.py

iProzd · Feb 28, 2024 · 75da5b1 · 75da5b1
1 parent 5a9df83
commit 75da5b1
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py
@@ -105,8 +105,6 @@ def __init__(
         self.model_keys = (
             list(model_params["model_dict"]) if self.multi_task else ["Default"]
         )
-        # if self.multi_task and sampled is None:
-        #     sampled = {key: None for key in self.model_keys}
         self.rank = dist.get_rank() if dist.is_initialized() else 0
         self.world_size = dist.get_world_size() if dist.is_initialized() else 1
         self.num_model = len(self.model_keys)