forked from wouterkool/attention-learn-to-route
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathargs.json
40 lines (40 loc) · 834 Bytes
/
args.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
{
"problem": "pctsp_stoch",
"graph_size": 50,
"batch_size": 512,
"epoch_size": 1280000,
"val_size": 10000,
"val_dataset": "",
"model": "attention",
"embedding_dim": 128,
"hidden_dim": 128,
"n_encode_layers": 3,
"tanh_clipping": 10.0,
"normalization": "batch",
"lr_model": 0.0001,
"lr_critic": 0.0001,
"lr_decay": 1.0,
"eval_only": false,
"n_epochs": 100,
"seed": 1234,
"max_grad_norm": 1.0,
"no_cuda": false,
"exp_beta": 0.8,
"baseline": "rollout",
"bl_alpha": 0.05,
"bl_warmup_epochs": 1,
"eval_batch_size": 2048,
"checkpoint_encoder": false,
"log_step": 50,
"log_dir": "logs",
"run_name": "pctsp_stoch_50",
"output_dir": "outputs",
"epoch_start": 0,
"checkpoint_epochs": 1,
"load_path": null,
"resume": null,
"no_tensorboard": false,
"no_progress_bar": false,
"use_cuda": true,
"save_dir": ""
}