add cvrp model

params.yaml

@@ -6,7 +6,7 @@ problem_params:
   PRICE_MODE: 1           # for pcvrp medic
   DISCOUNT_FACTOR: 0.9    # for pcvrp medic
   PLAYER_ROLE: engineer  # medic, engineer, maybe searcher?
-  MEDIC_MODEL_FILE: pretrained_asist/medic/pcvrp_medic_PM1_HV1
+  MEDIC_MODEL_FILE: pretrained_asist/medic/cvrp_baseline
   MEDIC_GRAPH_SIZE: 55        # number of victims (contains high value)
   RUBBLE_GRAPH_SIZE: 23       # number of rubbles
   HIGH_VALUE_VICTIM_SIZE: 5   # number of high value victims

pretrained_asist/medic/cvrp_baseline/args.json

@@ -0,0 +1,61 @@
+{
+ "arg_yaml_file": "params.yaml",
+ "problem": "cvrp",
+ "graph_size": 55,
+ "batch_size": 512,
+ "epoch_size": 12800,
+ "val_size": 10000,
+ "val_dataset": null,
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10,
+ "normalization": "batch",
+ "lr_model": 0.0001,
+ "lr_critic": 0.0001,
+ "lr_decay": 1.0,
+ "eval_only": false,
+ "n_epochs": 50,
+ "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": "saturn",
+ "log_step": 50,
+ "log_dir": "logs",
+ "run_name": "cvrp_baseline_20210723T133253",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 1,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "high_value": 1,
+ "cuda": [
+  1,
+  2,
+  3,
+  4,
+  5,
+  6,
+  7
+ ],
+ "problem_params": {
+  "PLAYER_ROLE": 1,
+  "PRICE_MODE": 1,
+  "DISCOUNT_FACTOR": 0.9,
+  "TOOL_DURABILITY": 20
+ },
+ "metric_file": "scores/scores.json",
+ "no_wandb": false,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_55/cvrp_baseline_20210723T133253"
+}