RNN-T update constants

Author: mwawrzos

mlperf_logging/mllog/constants.py

@@ -83,10 +83,7 @@
 LARS_OPT_END_LR = "lars_opt_end_learning_rate"
 LARS_OPT_LR_DECAY_POLY_POWER = "lars_opt_learning_rate_decay_poly_power"
 LARS_OPT_LR_DECAY_STEPS = "lars_opt_learning_rate_decay_steps"
-LARS_OPT_LR_MIN = "lars_opt_learning_rate_min"
 LARS_OPT_WEIGHT_DECAY = "lars_opt_weight_decay"
-LARS_OPT_BETA_1 = "lars_beta_1"
-LARS_OPT_BETA_2 = "lars_beta_2"
 MAX_IMAGE_SIZE = "max_image_size"
 MAX_SAMPLES = "max_samples"
 MAX_SEQUENCE_LENGTH = "max_sequence_length"
@@ -98,6 +95,13 @@
 OPT_ADAM_EPSILON = "opt_adam_epsilon"
 OPT_NAME = "opt_name"
 OPT_BASE_LR = "opt_base_learning_rate"
+OPT_LAMB_LR_MIN = "opt_lamb_learning_rate_min"
+OPT_LAMB_LR_DECAY_POLY_POWER = "opt_lamb_learning_rate_decay_poly_power"
+OPT_LAMB_BETA_1 = "opt_lamb_beta_1"
+OPT_LAMB_BETA_2 = "opt_lamb_beta_2"
+OPT_LAMB_EPSILON = "opt_lamb_epsilon"
+OPT_LAMB_LR_HOLD_EPOCHS = "opt_lamb_learning_rate_hold_epochs"
+OPT_LAMB_LR_WARMUP_EPOCHS = "opt_lamb_learning_rate_warmup_epochs"
 OPT_LR_ALT_DECAY_FUNC = "opt_learning_rate_alt_decay_func"
 OPT_LR_ALT_WARMUP_FUNC = "opt_learning_rate_alt_warmup_func"
 OPT_LR_DECAY_BOUNDARY_EPOCHS = "opt_learning_rate_decay_boundary_epochs"
@@ -110,7 +114,6 @@
 OPT_LR_WARMUP_EPOCHS = "opt_learning_rate_warmup_epochs"
 OPT_LR_WARMUP_FACTOR = "opt_learning_rate_warmup_factor"
 OPT_LR_WARMUP_STEPS = "opt_learning_rate_warmup_steps"
-OPT_LR_HOLD_EPOCHS = "opt_learning_rate_hold_epochs"
 OPT_WEIGHT_DECAY = "opt_weight_decay"
 OPT_GRADIENT_CLIP_NORM = "opt_gradient_clip_norm"
 DATA_SPEED_PERTURBATON_MAX = "data_speed_perturbaton_max"
@@ -121,8 +124,11 @@
 DATA_SPEC_AUGMENT_TIME_N = "data_spec_augment_time_n"
 DATA_SPEC_AUGMENT_TIME_MIN = "data_spec_augment_time_min"
 DATA_SPEC_AUGMENT_TIME_MAX = "data_spec_augment_time_max"
+DATA_TRAIN_NUM_BUCKETS = "data_train_num_buckets"
+DATA_TRAIN_MAX_DURATION = "data_train_max_duration"
 DATA_NUM_BUCKETS = "data_num_buckets"
-EMA_FACTOR = "ema_factor"
+MODEL_EVAL_EMA_FACTOR = "model_eval_ema_factor"
+EVAL_MAX_PREDICTION_SYMBOLS = "eval_max_prediction_symbols"
 
 # Log keys - misc.
 BBOX = "bbox"