diff --git a/gradient/constants.py b/gradient/constants.py index 95fa54b0..abd9b65c 100644 --- a/gradient/constants.py +++ b/gradient/constants.py @@ -2,6 +2,7 @@ class ExperimentType: SINGLE_NODE = 1 GRPC_MULTI_NODE = 2 MPI_MULTI_NODE = 3 + HYPERPARAMETER_TUNING = 4 @classmethod def get_type_str(cls, state_int): @@ -9,6 +10,7 @@ def get_type_str(cls, state_int): 1: "single node", 2: "GRPC multi node", 3: "MPI multi node", + 4: "Hyperparameter tuning" } return type_strings.get(state_int, "undefined")