File size: 649 Bytes
e2b7617 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# class ModelParameters():
# def __init__(self):
# self.NUM_LAYERS = "num_layers"
# self.NUM_HEADS = "num_heads"
# self.DIM_ATTENTION = "dim_attention"
# self.DIM_FEEDFORWARD = "dim_feedforward"
# self.DIM_LSTM = "dim_lstm"
# self.DIM_EMBEDDING = "dim_embedding"
# self.DIM_OUTPUT = "dim_output"
# self.RATE_DROPOUT = "rate_dropout"
# return
#
NUM_LAYERS = "num_layers"
NUM_HEADS = "num_heads"
DIM_ATTENTION = "dim_attention"
DIM_FEEDFORWARD = "dim_feedforward"
DIM_LSTM = "dim_lstm"
DIM_EMBEDDING = "dim_embedding"
DIM_OUTPUT = "dim_output"
RATE_DROPOUT = "rate_dropout"
|