Use NotImplementedError not ValueError.

intelligent-machine-learning · Oct 24, 2023 · 1e7f79c · 1e7f79c
1 parent 929e635
commit 1e7f79c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlrover/trainer/torch/elastic/checkpoint.py b/dlrover/trainer/torch/elastic/checkpoint.py
@@ -170,7 +170,7 @@ def init_checkpoint_manager(
                 max_to_keep,
             )
         else:
-            raise ValueError(f"Not support model class {model}")
+            raise NotImplementedError(f"Not support model class {model}")
 
 
 class LocalCheckpointManger(CheckpointManger):