| # 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" | |