From a19a5d8226bd7a482ec3811686202bbabdca8739 Mon Sep 17 00:00:00 2001 From: BartoszCki Date: Wed, 10 Jul 2019 12:48:56 +0200 Subject: [PATCH] Add missing hyperparameter experiment type --- gradient/constants.py | 2 ++ 1 file changed, 2 insertions(+) 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")