minimario's picture
upload models
74a5439
raw
history blame
1.5 kB
{
"problem": "cvrp",
"graph_size": 20,
"batch_size": 512,
"epoch_size": 1280000,
"val_size": 10000,
"val_dataset": "data/vrp/vrp20_validation_seed4321.pkl",
"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": 1024,
"checkpoint_encoder": false,
"shrink_size": null,
"data_distribution": null,
"log_step": 50,
"log_dir": "logs_test",
"run_dir": "runs_test",
"run_name": "dec10_starts_20221210T165824",
"output_dir": "outputs_test",
"image_dir": "images",
"epoch_start": 0,
"checkpoint_epochs": 1,
"load_path": null,
"resume": null,
"no_tensorboard": false,
"no_progress_bar": false,
"use_symmetry_ll": false,
"supervise_symmetry": false,
"restrict_symmetry": false,
"angle_threshold": 0.5,
"alternate": false,
"alternate_freq": 5,
"plot_freq": 0,
"supervise_lambda": 0.01,
"alex_symmetric_loss": false,
"both_losses": false,
"plot_symmetric_ll": false,
"cap_50": 40,
"decay_mode": -1,
"num_equivariant_samples": 0,
"use_starts": true,
"use_cuda": true,
"variant_name": "org",
"save_dir": "outputs_test/cvrp_20/org/dec10_starts_20221210T165824",
"plot_dir": "images/cvrp_20/org/dec10_starts_20221210T165824"
}