Skip to content

Commit

Permalink
fix_scalar_autodilatedrnn
Browse files Browse the repository at this point in the history
  • Loading branch information
elephaint committed Oct 8, 2024
1 parent ec32f28 commit 2801f19
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion action_files/test_models/src/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,8 @@ def main(dataset: str = 'M3', group: str = 'Monthly') -> None:
'encoder_hidden_size': tune.choice([124]),
"max_steps": 300,
"val_check_steps": 100,
"random_seed": tune.choice([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]),}
"random_seed": tune.choice([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]),
"scaler_type": "minmax1"}

models = [
AutoDilatedRNN(h=horizon, loss=MAE(), config=config_drnn, num_samples=2, cpus=1),
Expand Down

0 comments on commit 2801f19

Please sign in to comment.