Skip to content

Commit 3e68dee

Browse files
authored
Merge pull request #41 from Paperspace/Add-Hyperparameter-experiment-type
Add missing hyperparameter experiment type
2 parents 61b4b84 + a19a5d8 commit 3e68dee

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

gradient/constants.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,15 @@ class ExperimentType:
22
SINGLE_NODE = 1
33
GRPC_MULTI_NODE = 2
44
MPI_MULTI_NODE = 3
5+
HYPERPARAMETER_TUNING = 4
56

67
@classmethod
78
def get_type_str(cls, state_int):
89
type_strings = {
910
1: "single node",
1011
2: "GRPC multi node",
1112
3: "MPI multi node",
13+
4: "Hyperparameter tuning"
1214
}
1315
return type_strings.get(state_int, "undefined")
1416

0 commit comments

Comments
 (0)