minimario's picture
add offline models
7b2f326
{
"labelpath": "expert_data/vrp100_solution_9999.pkl",
"problem": "cvrp",
"graph_size": 100,
"batch_size": 100,
"epoch_size": 1280000,
"val_size": 10000,
"val_dataset": "expert_data/vrp100_valid_problem_9998.pkl",
"train_dataset": "expert_data/vrp100_problem_9999.pkl",
"model": "attention",
"embedding_dim": 128,
"hidden_dim": 128,
"n_encode_layers": 3,
"tanh_clipping": 10.0,
"normalization": "batch",
"label_aug": true,
"consistancy_learning": false,
"lr_model": 0.0001,
"lr_critic": 0.0001,
"lr_decay": 1.0,
"eval_only": false,
"n_epochs": 500,
"seed": 1234,
"max_grad_norm": 1.0,
"no_cuda": false,
"exp_beta": 0.8,
"baseline": null,
"bl_alpha": 0.05,
"bl_warmup_epochs": 0,
"eval_batch_size": 1024,
"checkpoint_encoder": false,
"shrink_size": null,
"data_distribution": null,
"log_step": 50,
"log_dir": "logs",
"run_name": "run_20230116T152849",
"wandb_run_name": "vrp100_inputaug4",
"output_dir": "outputs",
"epoch_start": 0,
"checkpoint_epochs": 10,
"load_path": null,
"resume": null,
"no_tensorboard": false,
"no_progress_bar": false,
"num_equivariant_samples": 0,
"supervise_lambda": 0.01,
"num_input_augmentations": 4,
"use_cuda": true,
"save_dir": "outputs/cvrp_100/run_20230116T152849"
}