| { | |
| "prev_hidden_dims": 64, | |
| "curr_hidden_dims": 512, | |
| "time_embedding_dims": 8, | |
| "lstm_embedding_dims": 128, | |
| "dropout_rate": 0.2, | |
| "num_layers": 4, | |
| "num_heads": 4, | |
| "batch_size": 8, | |
| "learning_rate": "1e-4", | |
| "weight_decay": "1e-4", | |
| "grad_accum_steps": 16, | |
| "pitch_control_channels": [ | |
| 16, | |
| 32, | |
| 32 | |
| ], | |
| "attention_dims": 32, | |
| "context_dims": 32 | |
| } |