diff --git a/run_100_eq5/args.json b/run_100_eq5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..382d1fff15dace38782d7c5aace4e55f9bc027d3
--- /dev/null
+++ b/run_100_eq5/args.json
@@ -0,0 +1,49 @@
+{
+ "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": false,
+ "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_20221230T235447",
+ "wandb_run_name": "100_eq5",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 5,
+ "supervise_lambda": 0.01,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T235447"
+}
\ No newline at end of file
diff --git a/run_100_eq5/epoch-499.pt b/run_100_eq5/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..98c30d5b87198790b5ab8a996bc63ebbf3cb85b5
--- /dev/null
+++ b/run_100_eq5/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3817df654c383af73e332baa8f11855fee472ea2a209e8ff47f4aad0fb3550d6
+size 8383429
diff --git a/run_100_eq5_lambda5/args.json b/run_100_eq5_lambda5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..e993a1b0c0ec3b3515a80a9a56689062b0d1f007
--- /dev/null
+++ b/run_100_eq5_lambda5/args.json
@@ -0,0 +1,49 @@
+{
+ "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": false,
+ "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_20221231T000851",
+ "wandb_run_name": "100_eq5_lambda5",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 5,
+ "supervise_lambda": 5.0,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221231T000851"
+}
\ No newline at end of file
diff --git a/run_100_eq5_lambda5/epoch-499.pt b/run_100_eq5_lambda5/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..6742698b80c66ec616b663ffdffde8fc30f067a3
--- /dev/null
+++ b/run_100_eq5_lambda5/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b76ef231b0681911871d10ec9d6784fb237962a8b09f96a90efc875de01998a
+size 8383429
diff --git a/run_100_orig/args.json b/run_100_orig/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..0b60b9f19e712908b06219b9adaa13f943394905
--- /dev/null
+++ b/run_100_orig/args.json
@@ -0,0 +1,49 @@
+{
+ "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": false,
+ "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_20221230T235242",
+ "wandb_run_name": "100_orig",
+ "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,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T235242"
+}
\ No newline at end of file
diff --git a/run_100_orig/epoch-499.pt b/run_100_orig/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..74c5ab31776497d7ce5a2b624d684f92bdcb080f
--- /dev/null
+++ b/run_100_orig/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7d38a4b41cacb9414a5d3fc8aa00590afd0c8ff68e247a9c53061e60a7deee1f
+size 8383429
diff --git a/run_20221230T16/args.json b/run_20221230T16/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..b2a087bebae0325a384dedfb9336f6003f057167
--- /dev/null
+++ b/run_20221230T16/args.json
@@ -0,0 +1,48 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_2345.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T165510",
+ "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,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T165510"
+}
\ No newline at end of file
diff --git a/run_20221230T16/epoch-499.pt b/run_20221230T16/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..64291ceb2370e3ba7fbc2f6e259fd8787a0ad838
--- /dev/null
+++ b/run_20221230T16/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:52a88ed3afbafc33ddf4d5cf2584e513dc476cef1e67f72549d65c6fecdf9dcf
+size 8383429
diff --git a/run_20221230T1645/args.json b/run_20221230T1645/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..a53de6fc076683e95a63ab2497d157e01bb3e43a
--- /dev/null
+++ b/run_20221230T1645/args.json
@@ -0,0 +1,48 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_2345.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T164510",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 1,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 0,
+ "supervise_lambda": 0.01,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T164510"
+}
\ No newline at end of file
diff --git a/run_20_big_label_cons/args.json b/run_20_big_label_cons/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..fe98cc243e00b519195ced6f5a8293f389089312
--- /dev/null
+++ b/run_20_big_label_cons/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_semibig_solution_7777.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_semibig_problem_7777.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": true,
+ "consistancy_learning": true,
+ "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_20221231T131833",
+ "wandb_run_name": "20_big_label_cons",
+ "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,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221231T131833"
+}
\ No newline at end of file
diff --git a/run_20_big_label_cons/epoch-499.pt b/run_20_big_label_cons/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..db007d1c6acdd09cfc2d3b637d8e33880c906336
--- /dev/null
+++ b/run_20_big_label_cons/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:07e6826d38736cad2a90583b489978fd4249a7637dc3a54b16af559e7e2ac369
+size 8383429
diff --git a/run_20_big_label_cons_eq5_lambda5/args.json b/run_20_big_label_cons_eq5_lambda5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..7df051b7cb02537b55b584ec4539a72f75bff199
--- /dev/null
+++ b/run_20_big_label_cons_eq5_lambda5/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_semibig_solution_7777.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_semibig_problem_7777.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": true,
+ "consistancy_learning": true,
+ "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_20221231T131828",
+ "wandb_run_name": "20_big_label_cons_eq5_lambda5",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 5,
+ "supervise_lambda": 5.0,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221231T131828"
+}
\ No newline at end of file
diff --git a/run_20_big_label_cons_eq5_lambda5/epoch-499.pt b/run_20_big_label_cons_eq5_lambda5/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..150651222899bde7ef22802e284d3b828ba3557a
--- /dev/null
+++ b/run_20_big_label_cons_eq5_lambda5/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:db94d0459759abec3ea97006c3e1dfa9e2412f3c41c710d1df1f605623efcc2f
+size 8383429
diff --git a/run_20_eq50_lambda20/args.json b/run_20_eq50_lambda20/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..2ca8c0f366ff7c90e32b5f084d298a1210bd5d34
--- /dev/null
+++ b/run_20_eq50_lambda20/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221231T002410",
+ "wandb_run_name": "20_eq50_lambda20",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 50,
+ "supervise_lambda": 20.0,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221231T002410"
+}
\ No newline at end of file
diff --git a/run_20_eq50_lambda20/epoch-499.pt b/run_20_eq50_lambda20/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..09bb71391daf2278a8b6e794f1ffd9962f8ee307
--- /dev/null
+++ b/run_20_eq50_lambda20/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d41fe7b554d718e17baf6159d09edfc7d54523cb7b539c9f7f02d17cffbed91f
+size 8383429
diff --git a/run_20_eq5_lambda20/args.json b/run_20_eq5_lambda20/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..6e68e54650cf6a7f7406999e5468163e422ccfaa
--- /dev/null
+++ b/run_20_eq5_lambda20/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221231T002328",
+ "wandb_run_name": "20_eq5_lambda20",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 5,
+ "supervise_lambda": 20.0,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221231T002328"
+}
\ No newline at end of file
diff --git a/run_20_eq5_lambda20/epoch-499.pt b/run_20_eq5_lambda20/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..b4b8a3b6e551a456e8460e18483a3eabdbf7a6c8
--- /dev/null
+++ b/run_20_eq5_lambda20/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:03ee5b5079abbefc65a7c36ea042d0349822281cd5b4a82a969ceb9dd4a73fc4
+size 8383429
diff --git a/run_eq_5/args.json b/run_eq_5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..69303090519b855b54f14723bdf85fe14c5c056c
--- /dev/null
+++ b/run_eq_5/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_2345.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T171551",
+ "wandb_run_name": "eq_5",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 5,
+ "supervise_lambda": 0.01,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T171551"
+}
\ No newline at end of file
diff --git a/run_eq_50/args.json b/run_eq_50/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..bf66f4ed4cf4f7256226bba2e52c266191d224ad
--- /dev/null
+++ b/run_eq_50/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_2345.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T171606",
+ "wandb_run_name": "eq_50",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 50,
+ "supervise_lambda": 0.01,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T171606"
+}
\ No newline at end of file
diff --git a/run_eq_50_1/args.json b/run_eq_50_1/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..c9ed6914c94efb39b911edff39d14d07d060de0d
--- /dev/null
+++ b/run_eq_50_1/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T172942",
+ "wandb_run_name": "eq_50",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 50,
+ "supervise_lambda": 0.01,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T172942"
+}
\ No newline at end of file
diff --git a/run_eq_5_2/args.json b/run_eq_5_2/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..2bd0a52ec3bbdcc182a895121ee649e84cdb3213
--- /dev/null
+++ b/run_eq_5_2/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T172937",
+ "wandb_run_name": "eq_5",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 5,
+ "supervise_lambda": 0.01,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T172937"
+}
\ No newline at end of file
diff --git a/run_eq_5_5/args.json b/run_eq_5_5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..754f49421b570fe89e8fec9503241303d24a1df5
--- /dev/null
+++ b/run_eq_5_5/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T194559",
+ "wandb_run_name": "eq_5",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 5,
+ "supervise_lambda": 0.5,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T194559"
+}
\ No newline at end of file
diff --git a/run_eq_5_lambda1/args.json b/run_eq_5_lambda1/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..7be867c0987aef613d9898b6976e6aa3c592cc9e
--- /dev/null
+++ b/run_eq_5_lambda1/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T195101",
+ "wandb_run_name": "eq_5_lambda1",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 5,
+ "supervise_lambda": 1.0,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T195101"
+}
\ No newline at end of file
diff --git a/run_eq_5_lambda5/args.json b/run_eq_5_lambda5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..8764d058e63d8b95586c6092463f7e60a5620900
--- /dev/null
+++ b/run_eq_5_lambda5/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T195720",
+ "wandb_run_name": "eq_5_lambda5",
+ "output_dir": "outputs",
+ "epoch_start": 0,
+ "checkpoint_epochs": 10,
+ "load_path": null,
+ "resume": null,
+ "no_tensorboard": false,
+ "no_progress_bar": false,
+ "num_equivariant_samples": 5,
+ "supervise_lambda": 5.0,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T195720"
+}
\ No newline at end of file
diff --git a/run_orig/args.json b/run_orig/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..ce7014db049f6f226e43ada3a20c5a33655843e2
--- /dev/null
+++ b/run_orig/args.json
@@ -0,0 +1,49 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20221230T172932",
+ "wandb_run_name": "orig",
+ "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,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20221230T172932"
+}
\ No newline at end of file
diff --git a/run_vrp100_10_inputaug2/args.json b/run_vrp100_10_inputaug2/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..cf94fa7ab946c611a9c001b2066ff65215cc8ca6
--- /dev/null
+++ b/run_vrp100_10_inputaug2/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp100_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp100_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp100_10_problem_2345.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_20230118T162807",
+ "wandb_run_name": "vrp100_10_inputaug2",
+ "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": 2,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230118T162807"
+}
\ No newline at end of file
diff --git a/run_vrp100_10_inputaug2/epoch-499.pt b/run_vrp100_10_inputaug2/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..4fe7fe8b84b88dacbb08d0c16ebe3243d88e98ca
--- /dev/null
+++ b/run_vrp100_10_inputaug2/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:500734c2c3890461b1a16e55c66a663b0643e32310ce4de7052caf142971790d
+size 8383429
diff --git a/run_vrp100_10_inputaug4/args.json b/run_vrp100_10_inputaug4/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..adbfa6d61742fed16f03313ed1594720fc7ef0c2
--- /dev/null
+++ b/run_vrp100_10_inputaug4/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp100_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp100_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp100_10_problem_2345.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_20230118T162808",
+ "wandb_run_name": "vrp100_10_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_20230118T162808"
+}
\ No newline at end of file
diff --git a/run_vrp100_10_inputaug4/epoch-499.pt b/run_vrp100_10_inputaug4/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..4529391d74a7cd0d987bb39cf170d199796b7de6
--- /dev/null
+++ b/run_vrp100_10_inputaug4/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc9a243b2d820d1823b77138401ff1a4da4f85bcbbadbeb85c3b4324e693cb3b
+size 8383429
diff --git a/run_vrp100_10_inputaug6/args.json b/run_vrp100_10_inputaug6/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..992f621e00af4d29a7a10ad98ef714d9fc77e372
--- /dev/null
+++ b/run_vrp100_10_inputaug6/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp100_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp100_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp100_10_problem_2345.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_20230118T162809",
+ "wandb_run_name": "vrp100_10_inputaug6",
+ "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": 6,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230118T162809"
+}
\ No newline at end of file
diff --git a/run_vrp100_10_inputaug6/epoch-499.pt b/run_vrp100_10_inputaug6/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..1a13e875e5cddf043d859b15a83f3b5f8fc09297
--- /dev/null
+++ b/run_vrp100_10_inputaug6/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cfd3524f5c9241b6849fd237fc14808e6b4745ae99c1819f969039167472a81
+size 8383429
diff --git a/run_vrp100_10_inputaug8/args.json b/run_vrp100_10_inputaug8/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..e9d0abeb0f574646954b478da2b6f89797c261f0
--- /dev/null
+++ b/run_vrp100_10_inputaug8/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp100_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp100_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp100_10_problem_2345.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_20230118T162810",
+ "wandb_run_name": "vrp100_10_inputaug8",
+ "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": 8,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230118T162810"
+}
\ No newline at end of file
diff --git a/run_vrp100_10_inputaug8/epoch-499.pt b/run_vrp100_10_inputaug8/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..9cd153dd29b97d7eb5dc5ba53e3138251eaf235a
--- /dev/null
+++ b/run_vrp100_10_inputaug8/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d102839c659c40076a647d824ad26c3f438db9c0370c7c98df3897ecc9c978eb
+size 8383429
diff --git a/run_vrp100_inputaug2/args.json b/run_vrp100_inputaug2/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..5953ba6843c50d891af8cffb6cbe1ec335801a52
--- /dev/null
+++ b/run_vrp100_inputaug2/args.json
@@ -0,0 +1,50 @@
+{
+ "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_20230116T152847",
+ "wandb_run_name": "vrp100_inputaug2",
+ "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": 2,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T152847"
+}
\ No newline at end of file
diff --git a/run_vrp100_inputaug2/epoch-499.pt b/run_vrp100_inputaug2/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..66a9ffd0225e35190ccf67d5c571823f9aa2168d
--- /dev/null
+++ b/run_vrp100_inputaug2/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f0f0866813f71d214d81aca646af479d0222a04a3f85a83f51bcfcc154c6b8b
+size 8383429
diff --git a/run_vrp100_inputaug4/args.json b/run_vrp100_inputaug4/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..cd319057d37afc526f85c6cbe6c7e569f50e2276
--- /dev/null
+++ b/run_vrp100_inputaug4/args.json
@@ -0,0 +1,50 @@
+{
+ "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"
+}
\ No newline at end of file
diff --git a/run_vrp100_inputaug4/epoch-499.pt b/run_vrp100_inputaug4/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..0933d0238ab864011ce546ff4765da26e6f2f7e6
--- /dev/null
+++ b/run_vrp100_inputaug4/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:416b2d4b1977581fbc16b899323e318ab07df0051a3a047751617f4460928036
+size 8383429
diff --git a/run_vrp100_inputaug6/args.json b/run_vrp100_inputaug6/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..a65e53b54ddaf3c0f4f9ec84c82c687c019edaba
--- /dev/null
+++ b/run_vrp100_inputaug6/args.json
@@ -0,0 +1,50 @@
+{
+ "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_20230116T152955",
+ "wandb_run_name": "vrp100_inputaug6",
+ "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": 6,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T152955"
+}
\ No newline at end of file
diff --git a/run_vrp100_inputaug6/epoch-499.pt b/run_vrp100_inputaug6/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..a559bd15d98face6eeed0225f655a40376616639
--- /dev/null
+++ b/run_vrp100_inputaug6/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fab1471da47e4a6354988d276576d1069272be03955cce48ac22a05e44a81a07
+size 8383429
diff --git a/run_vrp100_inputaug8/args.json b/run_vrp100_inputaug8/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..dac8942072c06a4ccbdfb3dad4fb0200286f30ae
--- /dev/null
+++ b/run_vrp100_inputaug8/args.json
@@ -0,0 +1,50 @@
+{
+ "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_20230116T152851",
+ "wandb_run_name": "vrp100_inputaug8",
+ "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": 8,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T152851"
+}
\ No newline at end of file
diff --git a/run_vrp100_inputaug8/epoch-499.pt b/run_vrp100_inputaug8/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..a8b77ccb2e32f67a08d9f908643871a8ba0f5971
--- /dev/null
+++ b/run_vrp100_inputaug8/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:742c27eaf3439815cc4cbb3fe9ec1918d6a323fcb7bd5ee3043d7dc666bed739
+size 8383429
diff --git a/run_vrp100_orig_lr1e-3/args.json b/run_vrp100_orig_lr1e-3/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..183ff8c873b94e8db8cedfe23980fb6f6b9244df
--- /dev/null
+++ b/run_vrp100_orig_lr1e-3/args.json
@@ -0,0 +1,50 @@
+{
+ "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": false,
+ "consistancy_learning": false,
+ "lr_model": 0.001,
+ "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_20230116T142603",
+ "wandb_run_name": "vrp100_orig_lr1e-3",
+ "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_20230116T142603"
+}
\ No newline at end of file
diff --git a/run_vrp100_orig_lr1e-3/epoch-499.pt b/run_vrp100_orig_lr1e-3/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..40663ff912053c362996fcd4af888cc27ef692d2
--- /dev/null
+++ b/run_vrp100_orig_lr1e-3/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1ce27efd8681c3eb2fadb184dea0e309d5c6ffd792eb95c56b22cf82349f094
+size 8383429
diff --git a/run_vrp100_orig_lr1e-5/args.json b/run_vrp100_orig_lr1e-5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..a04fd649f98f2732a4758907d86cb8f4d2ea0142
--- /dev/null
+++ b/run_vrp100_orig_lr1e-5/args.json
@@ -0,0 +1,50 @@
+{
+ "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": false,
+ "consistancy_learning": false,
+ "lr_model": 1e-05,
+ "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_20230116T142408",
+ "wandb_run_name": "vrp100_orig_lr1e-5",
+ "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_20230116T142408"
+}
\ No newline at end of file
diff --git a/run_vrp100_orig_lr1e-5/epoch-499.pt b/run_vrp100_orig_lr1e-5/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..d1f2796589a29028a8b6a5d682a113c35333f43f
--- /dev/null
+++ b/run_vrp100_orig_lr1e-5/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ce012def3e1892eed698d496d7ef31a43428d5bacd41763e96e6bd6929838c52
+size 8383429
diff --git a/run_vrp100_orig_lr4e-5/args.json b/run_vrp100_orig_lr4e-5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..989150ab94828d082e74f3639e6cc4343433f914
--- /dev/null
+++ b/run_vrp100_orig_lr4e-5/args.json
@@ -0,0 +1,50 @@
+{
+ "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": false,
+ "consistancy_learning": false,
+ "lr_model": 4e-05,
+ "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_20230116T142406",
+ "wandb_run_name": "vrp100_orig_lr4e-5",
+ "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_20230116T142406"
+}
\ No newline at end of file
diff --git a/run_vrp100_orig_lr4e-5/epoch-499.pt b/run_vrp100_orig_lr4e-5/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..9507be517cc1592534598b914fe3404eccba69f4
--- /dev/null
+++ b/run_vrp100_orig_lr4e-5/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cdc9cad15fd4384568dead7127ae6589b248d35a346b6ad78e25fe5c04cb0a27
+size 8383429
diff --git a/run_vrp100_orig_lr5e-4/args.json b/run_vrp100_orig_lr5e-4/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..8095c420bba368e23e897bf0d356cc3350cc9d87
--- /dev/null
+++ b/run_vrp100_orig_lr5e-4/args.json
@@ -0,0 +1,50 @@
+{
+ "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": false,
+ "consistancy_learning": false,
+ "lr_model": 0.0005,
+ "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_20230116T142600",
+ "wandb_run_name": "vrp100_orig_lr5e-4",
+ "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_20230116T142600"
+}
\ No newline at end of file
diff --git a/run_vrp100_orig_lr5e-4/epoch-499.pt b/run_vrp100_orig_lr5e-4/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..15362dec8fed7425eab7ce3abf6bd1091730e001
--- /dev/null
+++ b/run_vrp100_orig_lr5e-4/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0fa35edf7d44dcea290e9dc1567b758b97f396f621e03d708d814fb77c86d7d7
+size 8383429
diff --git a/run_vrp100_orig_lr5e-6/args.json b/run_vrp100_orig_lr5e-6/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..176d5cf692242671c274e92e7d6d8291f6f825b3
--- /dev/null
+++ b/run_vrp100_orig_lr5e-6/args.json
@@ -0,0 +1,50 @@
+{
+ "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": false,
+ "consistancy_learning": false,
+ "lr_model": 5e-06,
+ "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_20230116T142410",
+ "wandb_run_name": "vrp100_orig_lr5e-6",
+ "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_20230116T142410"
+}
\ No newline at end of file
diff --git a/run_vrp100_orig_lr5e-6/epoch-499.pt b/run_vrp100_orig_lr5e-6/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..bddd343505eb94c8aa12cfc4fecab06f60f5a4f2
--- /dev/null
+++ b/run_vrp100_orig_lr5e-6/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ceaed33fb73207869eb9c620b0fc66b5d48b306b443c1698a09891e9ace8efe
+size 8383429
diff --git a/run_vrp100_orig_lr7e-5/args.json b/run_vrp100_orig_lr7e-5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..3568a33a8aba704a2571e0ea34870cb1be4351bb
--- /dev/null
+++ b/run_vrp100_orig_lr7e-5/args.json
@@ -0,0 +1,50 @@
+{
+ "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": false,
+ "consistancy_learning": false,
+ "lr_model": 7e-05,
+ "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_20230116T142404",
+ "wandb_run_name": "vrp100_orig_lr7e-5",
+ "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_20230116T142404"
+}
\ No newline at end of file
diff --git a/run_vrp100_orig_lr7e-5/epoch-499.pt b/run_vrp100_orig_lr7e-5/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..ae7d3eabbab683469ce5e712f318f5c5bbe41007
--- /dev/null
+++ b/run_vrp100_orig_lr7e-5/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:026b1593ed108b2825085b0efdae93b41a171cbf854618bc7e80229f1251dedf
+size 8383429
diff --git a/run_vrp100_tiny_inputaug2/args.json b/run_vrp100_tiny_inputaug2/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..94fe238f94aada67dc12ca7714c2406601c0c0ea
--- /dev/null
+++ b/run_vrp100_tiny_inputaug2/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp100_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp100_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp100_tiny_problem_2345.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_20230116T234311",
+ "wandb_run_name": "vrp100_tiny_inputaug2",
+ "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": 2,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T234311"
+}
\ No newline at end of file
diff --git a/run_vrp100_tiny_inputaug2/epoch-499.pt b/run_vrp100_tiny_inputaug2/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..b7e7baf2937cf294e2b4da41fdc10d4c57107ca7
--- /dev/null
+++ b/run_vrp100_tiny_inputaug2/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b7d2e12944ed5a458b0a33b7195aa3f79495e6d31461db90233dc209333b3aa9
+size 8383429
diff --git a/run_vrp100_tiny_inputaug4/args.json b/run_vrp100_tiny_inputaug4/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..ee56c1c0618c865454da0fbc27f5b0e362cddcb8
--- /dev/null
+++ b/run_vrp100_tiny_inputaug4/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp100_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp100_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp100_tiny_problem_2345.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_20230116T234313",
+ "wandb_run_name": "vrp100_tiny_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_20230116T234313"
+}
\ No newline at end of file
diff --git a/run_vrp100_tiny_inputaug4/epoch-499.pt b/run_vrp100_tiny_inputaug4/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..cde243f1e05a53e87470b2ff998a3a1ba0984c1a
--- /dev/null
+++ b/run_vrp100_tiny_inputaug4/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:46939c207f3c9d3f09f0ebe3840b932c4d018422ad90eb3938f8665e9d533ab3
+size 8383429
diff --git a/run_vrp100_tiny_inputaug8/args.json b/run_vrp100_tiny_inputaug8/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..7999856b7ce93ee940bc8607a26db96539e225c4
--- /dev/null
+++ b/run_vrp100_tiny_inputaug8/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp100_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp100_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp100_tiny_problem_2345.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_20230116T234315",
+ "wandb_run_name": "vrp100_tiny_inputaug8",
+ "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": 8,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T234315"
+}
\ No newline at end of file
diff --git a/run_vrp100_tiny_inputaug8/epoch-499.pt b/run_vrp100_tiny_inputaug8/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..7b314f9930564373ed8bdfaaaa3f32b226755f20
--- /dev/null
+++ b/run_vrp100_tiny_inputaug8/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:499df018c282cd3a37c5ddff39c23ebb1ba129fdafdc354e32682159229c9d05
+size 8383429
diff --git a/run_vrp20_10_inputaug16/args.json b/run_vrp20_10_inputaug16/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..de8d30533c2756a8e3ba35b3da0be40305d89c9a
--- /dev/null
+++ b/run_vrp20_10_inputaug16/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_10_problem_2345.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_20230117T130341",
+ "wandb_run_name": "vrp20_10_inputaug16",
+ "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": 16,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T130341"
+}
\ No newline at end of file
diff --git a/run_vrp20_10_inputaug16/epoch-499.pt b/run_vrp20_10_inputaug16/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..1f17796ebecfb395e49471e0bebb1410d3e50260
--- /dev/null
+++ b/run_vrp20_10_inputaug16/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:979cc4727428f7beae9fdbbbba4b707ae78c7fc831516f43aa6217cd3f729142
+size 8383429
diff --git a/run_vrp20_10_inputaug2/args.json b/run_vrp20_10_inputaug2/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..e82e6adfba03c03a11661135c8affb4acd8d157a
--- /dev/null
+++ b/run_vrp20_10_inputaug2/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_10_problem_2345.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_20230117T130337",
+ "wandb_run_name": "vrp20_10_inputaug2",
+ "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": 2,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T130337"
+}
\ No newline at end of file
diff --git a/run_vrp20_10_inputaug2/epoch-499.pt b/run_vrp20_10_inputaug2/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..18bf48680e9ec31aa9c30e59bed12b35a7129dac
--- /dev/null
+++ b/run_vrp20_10_inputaug2/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8ad7100800c38faaa668d3a3833129fa50052afbaf0d40e76b5867faa2a50462
+size 8383429
diff --git a/run_vrp20_10_inputaug32/args.json b/run_vrp20_10_inputaug32/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..f798cad8f1f178dfca71558bfe49b8877fea181f
--- /dev/null
+++ b/run_vrp20_10_inputaug32/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_10_problem_2345.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_20230117T130342",
+ "wandb_run_name": "vrp20_10_inputaug32",
+ "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": 32,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T130342"
+}
\ No newline at end of file
diff --git a/run_vrp20_10_inputaug32/epoch-499.pt b/run_vrp20_10_inputaug32/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..a75e7c9f5b0023aafb7f548008165badb5deee6a
--- /dev/null
+++ b/run_vrp20_10_inputaug32/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:23e78126d0f8f5f0b6b105a1a23c1c0949aab15c6af1a9073c419a987a30d03a
+size 8383429
diff --git a/run_vrp20_10_inputaug32_lr0.000005/args.json b/run_vrp20_10_inputaug32_lr0.000005/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..6da1c4df3558580be21972903b5901485f257d2c
--- /dev/null
+++ b/run_vrp20_10_inputaug32_lr0.000005/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_10_problem_2345.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": 5e-06,
+ "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_20230117T131714",
+ "wandb_run_name": "vrp20_10_inputaug32_lr0.000005",
+ "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": 32,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T131714"
+}
\ No newline at end of file
diff --git a/run_vrp20_10_inputaug32_lr0.000005/epoch-499.pt b/run_vrp20_10_inputaug32_lr0.000005/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..290063410095852251d8e448064ce583a77c09b4
--- /dev/null
+++ b/run_vrp20_10_inputaug32_lr0.000005/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1b31abd1fa93bb2ff35d814748e6a8fb944dcc8e5f01706c25ecdcf8094ee410
+size 8383429
diff --git a/run_vrp20_10_inputaug32_lr0.00001/args.json b/run_vrp20_10_inputaug32_lr0.00001/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..868436d6ca080a024054608c714ee7ffcd7eac30
--- /dev/null
+++ b/run_vrp20_10_inputaug32_lr0.00001/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_10_problem_2345.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": 1e-05,
+ "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_20230117T131715",
+ "wandb_run_name": "vrp20_10_inputaug32_lr0.00001",
+ "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": 32,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T131715"
+}
\ No newline at end of file
diff --git a/run_vrp20_10_inputaug32_lr0.00001/epoch-499.pt b/run_vrp20_10_inputaug32_lr0.00001/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..e00c265e769245170193f51afaee1828c771e690
--- /dev/null
+++ b/run_vrp20_10_inputaug32_lr0.00001/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6fb85e96e407233ea4a9b6062f48c307d7f4959cb2289e7e3dfe8e39a528109b
+size 8383429
diff --git a/run_vrp20_10_inputaug32_lr0.00005/args.json b/run_vrp20_10_inputaug32_lr0.00005/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..f13f8167113a478ce343c627af8a05ba948cd9a0
--- /dev/null
+++ b/run_vrp20_10_inputaug32_lr0.00005/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_10_problem_2345.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": 5e-05,
+ "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_20230117T131716",
+ "wandb_run_name": "vrp20_10_inputaug32_lr0.00005",
+ "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": 32,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T131716"
+}
\ No newline at end of file
diff --git a/run_vrp20_10_inputaug32_lr0.00005/epoch-499.pt b/run_vrp20_10_inputaug32_lr0.00005/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..d5c0258ff1f940d472344cb40a4ba2533243bded
--- /dev/null
+++ b/run_vrp20_10_inputaug32_lr0.00005/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5e3bbc2dd7424e7d85b952166b4365d222fd6d4049b805f39ba3e269ebc504b0
+size 8383429
diff --git a/run_vrp20_10_inputaug32_lr0.0005/args.json b/run_vrp20_10_inputaug32_lr0.0005/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..018c33a7d2c80d83ec7588907678aa7c81a6f100
--- /dev/null
+++ b/run_vrp20_10_inputaug32_lr0.0005/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_10_problem_2345.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.0005,
+ "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_20230117T131717",
+ "wandb_run_name": "vrp20_10_inputaug32_lr0.0005",
+ "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": 32,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T131717"
+}
\ No newline at end of file
diff --git a/run_vrp20_10_inputaug32_lr0.0005/epoch-499.pt b/run_vrp20_10_inputaug32_lr0.0005/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..b1b678d20f5056db6024b55695155729c87736e2
--- /dev/null
+++ b/run_vrp20_10_inputaug32_lr0.0005/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:02e59de7a53261252b77a81f5155dd5715c55f1d888c61edfe908962832f3807
+size 8383429
diff --git a/run_vrp20_10_inputaug4/args.json b/run_vrp20_10_inputaug4/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..aea57f98919b1c9a6c89454ef81ffb11c99194e0
--- /dev/null
+++ b/run_vrp20_10_inputaug4/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_10_problem_2345.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_20230117T130338",
+ "wandb_run_name": "vrp20_10_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_20230117T130338"
+}
\ No newline at end of file
diff --git a/run_vrp20_10_inputaug4/epoch-499.pt b/run_vrp20_10_inputaug4/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..2c3ac000d3f12d2e1f4975b7bad28dc9842e9ec8
--- /dev/null
+++ b/run_vrp20_10_inputaug4/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bbc4568f8ee90f7c408723d5567350e5b904af1c2bee37700e36590e9c073acf
+size 8383429
diff --git a/run_vrp20_10_inputaug8/args.json b/run_vrp20_10_inputaug8/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..cceab4cc87bcb0f20e74d2dc19a191ca9698698f
--- /dev/null
+++ b/run_vrp20_10_inputaug8/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_10_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_10_problem_2345.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_20230117T130340",
+ "wandb_run_name": "vrp20_10_inputaug8",
+ "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": 8,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T130340"
+}
\ No newline at end of file
diff --git a/run_vrp20_10_inputaug8/epoch-499.pt b/run_vrp20_10_inputaug8/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..1fd26cb7affa0ccc2b1b6f1e2b02c155658e1616
--- /dev/null
+++ b/run_vrp20_10_inputaug8/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:12db745f5eecd323a87a857892dd90b07764c68aca5abeee0144f833cf00b143
+size 8383429
diff --git a/run_vrp20_inputaug2/args.json b/run_vrp20_inputaug2/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..0bfc8b03fbddd4a3aad982bcb107772038d748ee
--- /dev/null
+++ b/run_vrp20_inputaug2/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.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_20230116T185157",
+ "wandb_run_name": "vrp20_inputaug2",
+ "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": 2,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T185157"
+}
\ No newline at end of file
diff --git a/run_vrp20_inputaug2/epoch-499.pt b/run_vrp20_inputaug2/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..e0c8fbbb1c72b105b74a74b28243f27809bb6293
--- /dev/null
+++ b/run_vrp20_inputaug2/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0272db399d489f6f9b625f040f469573f368ff48493b5c645aae270752c103bc
+size 8383429
diff --git a/run_vrp20_inputaug32/args.json b/run_vrp20_inputaug32/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..b3e80fcdc5253bd07093d10d9e04c4179b24d0f5
--- /dev/null
+++ b/run_vrp20_inputaug32/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.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_20230116T185205",
+ "wandb_run_name": "vrp20_inputaug32",
+ "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": 32,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T185205"
+}
\ No newline at end of file
diff --git a/run_vrp20_inputaug32/epoch-499.pt b/run_vrp20_inputaug32/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..266a84fa959be619ff909f30944c7fdf454c1342
--- /dev/null
+++ b/run_vrp20_inputaug32/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:216d5997585877392607fb443982ec3e80911f5d4f45d740bbdd75e59d970c00
+size 8383429
diff --git a/run_vrp20_inputaug4/args.json b/run_vrp20_inputaug4/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..bf472653f964b3c109159db7b67cba4f7d74f246
--- /dev/null
+++ b/run_vrp20_inputaug4/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.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_20230116T185159",
+ "wandb_run_name": "vrp20_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_20230116T185159"
+}
\ No newline at end of file
diff --git a/run_vrp20_inputaug4/epoch-499.pt b/run_vrp20_inputaug4/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..50a5aa7221f3247495ce2a55ff3d0c8fab54dfe4
--- /dev/null
+++ b/run_vrp20_inputaug4/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76da7d93bdfd3e3f6e70ff9b9a398f2111c4e46deea308977734be1a0da145d1
+size 8383429
diff --git a/run_vrp20_inputaug6/args.json b/run_vrp20_inputaug6/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..da8d66621759e404f307a41f0fa81659e41c09f0
--- /dev/null
+++ b/run_vrp20_inputaug6/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.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_20230116T185203",
+ "wandb_run_name": "vrp20_inputaug6",
+ "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": 16,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T185203"
+}
\ No newline at end of file
diff --git a/run_vrp20_inputaug6/epoch-499.pt b/run_vrp20_inputaug6/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..be98815e44805374718d9606edd83bcaf8304170
--- /dev/null
+++ b/run_vrp20_inputaug6/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d8428340cb3736b780ba6dab8af5d2841ace39f9047f44902368254be137042e
+size 8383429
diff --git a/run_vrp20_inputaug64/args.json b/run_vrp20_inputaug64/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..6a61e7a6eaec0e290ea754da7d61343189e5e6ba
--- /dev/null
+++ b/run_vrp20_inputaug64/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.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_20230116T185207",
+ "wandb_run_name": "vrp20_inputaug64",
+ "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": 64,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T185207"
+}
\ No newline at end of file
diff --git a/run_vrp20_inputaug64/epoch-499.pt b/run_vrp20_inputaug64/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..ee39dd1a2c61f3c98d784cbd41c9992bcea11e45
--- /dev/null
+++ b/run_vrp20_inputaug64/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c960f66576eefdc5eb06a57dcdf56665ef2c8563b4e644fdc1dd694a80940cb5
+size 8383429
diff --git a/run_vrp20_inputaug8/args.json b/run_vrp20_inputaug8/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..f3960b5ac4b29fe6af84dce8384312db0b4b75c0
--- /dev/null
+++ b/run_vrp20_inputaug8/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_problem_2345.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_20230116T185201",
+ "wandb_run_name": "vrp20_inputaug8",
+ "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": 8,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T185201"
+}
\ No newline at end of file
diff --git a/run_vrp20_inputaug8/epoch-499.pt b/run_vrp20_inputaug8/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..5f41f111a6485d1aab4f06d3f19170b0cc8c7d88
--- /dev/null
+++ b/run_vrp20_inputaug8/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:13a7ee75371207a8a8e022e11a60e184bef3a286a297bde5e5d6cbd43c5dc530
+size 8383429
diff --git a/run_vrp20_normal_inputaug16/args.json b/run_vrp20_normal_inputaug16/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..7dc94f7151974c9bd6385585a30b5df42b02a990
--- /dev/null
+++ b/run_vrp20_normal_inputaug16/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_normal_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_normal_problem_2345.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_20230117T145030",
+ "wandb_run_name": "vrp20_normal_inputaug16",
+ "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": 16,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T145030"
+}
\ No newline at end of file
diff --git a/run_vrp20_normal_inputaug2/args.json b/run_vrp20_normal_inputaug2/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..54f33ef8cf4fcc3684882a15cb7333b22a76bf87
--- /dev/null
+++ b/run_vrp20_normal_inputaug2/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_normal_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_normal_problem_2345.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_20230117T145025",
+ "wandb_run_name": "vrp20_normal_inputaug2",
+ "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": 2,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T145025"
+}
\ No newline at end of file
diff --git a/run_vrp20_normal_inputaug3/args.json b/run_vrp20_normal_inputaug3/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..f3857202af15a327df06f86f59946e0e932c3d07
--- /dev/null
+++ b/run_vrp20_normal_inputaug3/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_normal_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_normal_problem_2345.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_20230117T145026",
+ "wandb_run_name": "vrp20_normal_inputaug3",
+ "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": 3,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T145026"
+}
\ No newline at end of file
diff --git a/run_vrp20_normal_inputaug32/args.json b/run_vrp20_normal_inputaug32/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..b8e4059fd1ce2a61b532dffb8aa4d409d2fdfb00
--- /dev/null
+++ b/run_vrp20_normal_inputaug32/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_normal_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_normal_problem_2345.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_20230117T145031",
+ "wandb_run_name": "vrp20_normal_inputaug32",
+ "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": 32,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T145031"
+}
\ No newline at end of file
diff --git a/run_vrp20_normal_inputaug4/args.json b/run_vrp20_normal_inputaug4/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..18139780d1fd7950a8e1f7106de98e74d055f9c8
--- /dev/null
+++ b/run_vrp20_normal_inputaug4/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_normal_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_normal_problem_2345.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_20230117T145027",
+ "wandb_run_name": "vrp20_normal_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_20230117T145027"
+}
\ No newline at end of file
diff --git a/run_vrp20_normal_inputaug6/args.json b/run_vrp20_normal_inputaug6/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..bf63a6ae531a8447a9c48561cc1b3969310c0d90
--- /dev/null
+++ b/run_vrp20_normal_inputaug6/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_normal_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_normal_problem_2345.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_20230117T145028",
+ "wandb_run_name": "vrp20_normal_inputaug6",
+ "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": 6,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T145028"
+}
\ No newline at end of file
diff --git a/run_vrp20_normal_inputaug8/args.json b/run_vrp20_normal_inputaug8/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..86f5c1456e7789cf24f5638bc169f05ff0471c71
--- /dev/null
+++ b/run_vrp20_normal_inputaug8/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_normal_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_normal_problem_2345.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_20230117T145029",
+ "wandb_run_name": "vrp20_normal_inputaug8",
+ "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": 8,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T145029"
+}
\ No newline at end of file
diff --git a/run_vrp20_small_inputaug16/args.json b/run_vrp20_small_inputaug16/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..edac26254189a9efd1248a176e06b92eb770ef87
--- /dev/null
+++ b/run_vrp20_small_inputaug16/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_small_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_small_problem_2345.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_20230116T225603",
+ "wandb_run_name": "vrp20_small_inputaug16",
+ "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": 16,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T225603"
+}
\ No newline at end of file
diff --git a/run_vrp20_small_inputaug16/epoch-499.pt b/run_vrp20_small_inputaug16/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..c8007db5e204d77b08e9d7f842ea53dd1f5df161
--- /dev/null
+++ b/run_vrp20_small_inputaug16/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed133c0691cd78b61641cac731b3d31b5f6860452fc67144f10dfe30c73b56e7
+size 8383429
diff --git a/run_vrp20_small_inputaug2/args.json b/run_vrp20_small_inputaug2/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..dde3b9672b4b351bd6734b51ec7557c259b9e7e9
--- /dev/null
+++ b/run_vrp20_small_inputaug2/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_small_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_small_problem_2345.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_20230117T125148",
+ "wandb_run_name": "vrp20_small_inputaug2",
+ "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": 2,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230117T125148"
+}
\ No newline at end of file
diff --git a/run_vrp20_small_inputaug2_5/args.json b/run_vrp20_small_inputaug2_5/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..9f7686a5b71d4ac858aab7f93eaaeaeb84effd35
--- /dev/null
+++ b/run_vrp20_small_inputaug2_5/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_small_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_small_problem_2345.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_20230116T225556",
+ "wandb_run_name": "vrp20_small_inputaug2",
+ "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": 2,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T225556"
+}
\ No newline at end of file
diff --git a/run_vrp20_small_inputaug2_5/epoch-499.pt b/run_vrp20_small_inputaug2_5/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..dfe629f9e1889774f0aab7c7be83c6e7a1a1ad7d
--- /dev/null
+++ b/run_vrp20_small_inputaug2_5/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e0e154965e071a5ea699c5405952f2ce79b6ee8705de98f791f7ddeeee473259
+size 8383429
diff --git a/run_vrp20_small_inputaug32/args.json b/run_vrp20_small_inputaug32/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..9fc26a4f3aecc0a28467cda4074b6f45f2ec18d8
--- /dev/null
+++ b/run_vrp20_small_inputaug32/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_small_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_small_problem_2345.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_20230116T225605",
+ "wandb_run_name": "vrp20_small_inputaug32",
+ "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": 32,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T225605"
+}
\ No newline at end of file
diff --git a/run_vrp20_small_inputaug32/epoch-499.pt b/run_vrp20_small_inputaug32/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..a889ca316b601a5762d87a45f7c7be104b247d29
--- /dev/null
+++ b/run_vrp20_small_inputaug32/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c2687614a3b70bbfa9a63f0558ffe3ec5f93ec8b5458f08ee4f3a77036424efa
+size 8383429
diff --git a/run_vrp20_small_inputaug4/args.json b/run_vrp20_small_inputaug4/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..9af690a47f75ba6e5d3becda445fdf64368df6f0
--- /dev/null
+++ b/run_vrp20_small_inputaug4/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_small_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_small_problem_2345.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_20230116T225558",
+ "wandb_run_name": "vrp20_small_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_20230116T225558"
+}
\ No newline at end of file
diff --git a/run_vrp20_small_inputaug4/epoch-499.pt b/run_vrp20_small_inputaug4/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..fd41ae9cef7c600192e8505d8da6257896a435d8
--- /dev/null
+++ b/run_vrp20_small_inputaug4/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c656ac29c2a9362495b4b71d4ea68063204b95f185c9c91f82b3d81aec4cb3a
+size 8383429
diff --git a/run_vrp20_small_inputaug64/args.json b/run_vrp20_small_inputaug64/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..d8ba912e05580ea6781743837a8acc13253d3993
--- /dev/null
+++ b/run_vrp20_small_inputaug64/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_small_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_small_problem_2345.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_20230116T225607",
+ "wandb_run_name": "vrp20_small_inputaug64",
+ "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": 64,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T225607"
+}
\ No newline at end of file
diff --git a/run_vrp20_small_inputaug64/epoch-499.pt b/run_vrp20_small_inputaug64/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..1bf012d6f8f82373a7a882e8805f0b6a66e62b72
--- /dev/null
+++ b/run_vrp20_small_inputaug64/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9b5a9f5e4d7469fc36215ce80f411bd9775952811b319941300f3804de9d331f
+size 8383429
diff --git a/run_vrp20_small_inputaug8/args.json b/run_vrp20_small_inputaug8/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..fcc868e1c57e1369cbff6d94352a1ec934fca3f6
--- /dev/null
+++ b/run_vrp20_small_inputaug8/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_small_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_small_problem_2345.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_20230116T225601",
+ "wandb_run_name": "vrp20_small_inputaug8",
+ "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": 8,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T225601"
+}
\ No newline at end of file
diff --git a/run_vrp20_small_inputaug8/epoch-499.pt b/run_vrp20_small_inputaug8/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..c4d6b3f7bb5a15463ca2e8ee31847f09d3e0b4cb
--- /dev/null
+++ b/run_vrp20_small_inputaug8/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ff4710c414ea5ca60652de7ab5b053c071b10f3951b6b4076523ec119807cf4f
+size 8383429
diff --git a/run_vrp20_tiny_inputaug16/args.json b/run_vrp20_tiny_inputaug16/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..d5815cb2097ac8d0f8c971491c23259d6ab775e3
--- /dev/null
+++ b/run_vrp20_tiny_inputaug16/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_tiny_problem_2345.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_20230116T233501",
+ "wandb_run_name": "vrp20_tiny_inputaug16",
+ "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": 16,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T233501"
+}
\ No newline at end of file
diff --git a/run_vrp20_tiny_inputaug16/epoch-499.pt b/run_vrp20_tiny_inputaug16/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..e1b01d528aea980867849905aa5e3b95c57a146b
--- /dev/null
+++ b/run_vrp20_tiny_inputaug16/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:44a946f6a1862f4aebacd085dc64873afcc01572538387f0470400c820d6769d
+size 8383429
diff --git a/run_vrp20_tiny_inputaug2/args.json b/run_vrp20_tiny_inputaug2/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..4578359dfaefdfb66ef84a5c774ddf998080a9fc
--- /dev/null
+++ b/run_vrp20_tiny_inputaug2/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_tiny_problem_2345.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_20230116T233455",
+ "wandb_run_name": "vrp20_tiny_inputaug2",
+ "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": 2,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T233455"
+}
\ No newline at end of file
diff --git a/run_vrp20_tiny_inputaug2/epoch-499.pt b/run_vrp20_tiny_inputaug2/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..f22b7affe075086869e0386ee01a2021d9589699
--- /dev/null
+++ b/run_vrp20_tiny_inputaug2/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f060a2f5be07564f872abb2f0df75971e4af3e5f2762ff4dee46f9d6a45fddf0
+size 8383429
diff --git a/run_vrp20_tiny_inputaug32/args.json b/run_vrp20_tiny_inputaug32/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..4b145f8d62cf4af25ea6f88d4241fbcc07f80cd1
--- /dev/null
+++ b/run_vrp20_tiny_inputaug32/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_tiny_problem_2345.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_20230116T233503",
+ "wandb_run_name": "vrp20_tiny_inputaug32",
+ "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": 32,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T233503"
+}
\ No newline at end of file
diff --git a/run_vrp20_tiny_inputaug32/epoch-499.pt b/run_vrp20_tiny_inputaug32/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..9f651e47a0bace46983aacba40a8eeac9a8cc570
--- /dev/null
+++ b/run_vrp20_tiny_inputaug32/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0efc22e47727c915928aadaf126c5f525b89d1fae90a44f8be5acb7ce091211c
+size 8383429
diff --git a/run_vrp20_tiny_inputaug4/args.json b/run_vrp20_tiny_inputaug4/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..0c1a1f32babd4bdb99230f05a35ebf6e61693c8a
--- /dev/null
+++ b/run_vrp20_tiny_inputaug4/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_tiny_problem_2345.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_20230116T233457",
+ "wandb_run_name": "vrp20_tiny_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_20230116T233457"
+}
\ No newline at end of file
diff --git a/run_vrp20_tiny_inputaug4/epoch-499.pt b/run_vrp20_tiny_inputaug4/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..9f563b0eeaa02b2b15c083ec0cb26c2b41e11730
--- /dev/null
+++ b/run_vrp20_tiny_inputaug4/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65f4684e05c669f121d5bc55bced8ee5ede476c5f21401d8d4ae7cb32173c002
+size 8383429
diff --git a/run_vrp20_tiny_inputaug64/args.json b/run_vrp20_tiny_inputaug64/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..f063aea86ad4924d16d689f7157921d5b5e52b71
--- /dev/null
+++ b/run_vrp20_tiny_inputaug64/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_tiny_problem_2345.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_20230116T233505",
+ "wandb_run_name": "vrp20_tiny_inputaug64",
+ "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": 64,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T233505"
+}
\ No newline at end of file
diff --git a/run_vrp20_tiny_inputaug64/epoch-499.pt b/run_vrp20_tiny_inputaug64/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..66b1819033c9dd12973a51227a8f834a9729d72a
--- /dev/null
+++ b/run_vrp20_tiny_inputaug64/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a96f7b30355d586982a8e4f4df440e3d446c960d23673d547331ccd29652d132
+size 8383429
diff --git a/run_vrp20_tiny_inputaug8/args.json b/run_vrp20_tiny_inputaug8/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..15e897b45ccc94add8fd03c50000589b56e56b50
--- /dev/null
+++ b/run_vrp20_tiny_inputaug8/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_tiny_problem_2345.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_20230116T233459",
+ "wandb_run_name": "vrp20_tiny_inputaug8",
+ "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": 8,
+ "use_cuda": true,
+ "save_dir": "outputs/cvrp_100/run_20230116T233459"
+}
\ No newline at end of file
diff --git a/run_vrp20_tiny_inputaug8/epoch-499.pt b/run_vrp20_tiny_inputaug8/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..5c50e117e70d424f37ab6d7506df25eac44527f8
--- /dev/null
+++ b/run_vrp20_tiny_inputaug8/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64e3acfe862d80ee26de07408fb397e4b1e86676d3d51003acafdedaac78eddd
+size 8383429
diff --git a/run_vrp20_tiny_orig/args.json b/run_vrp20_tiny_orig/args.json
new file mode 100644
index 0000000000000000000000000000000000000000..a1788a45e2c694c42a781f1dc2c9b6d567da0ded
--- /dev/null
+++ b/run_vrp20_tiny_orig/args.json
@@ -0,0 +1,50 @@
+{
+ "labelpath": "expert_data/vrp20_tiny_solution_2345.pkl",
+ "problem": "cvrp",
+ "graph_size": 100,
+ "batch_size": 100,
+ "epoch_size": 1280000,
+ "val_size": 10000,
+ "val_dataset": "expert_data/vrp20_valid_problem_3456.pkl",
+ "train_dataset": "expert_data/vrp20_tiny_problem_2345.pkl",
+ "model": "attention",
+ "embedding_dim": 128,
+ "hidden_dim": 128,
+ "n_encode_layers": 3,
+ "tanh_clipping": 10.0,
+ "normalization": "batch",
+ "label_aug": false,
+ "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_20230117T143819",
+ "wandb_run_name": "vrp20_tiny_orig",
+ "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_20230117T143819"
+}
\ No newline at end of file
diff --git a/run_vrp20_tiny_orig/epoch-499.pt b/run_vrp20_tiny_orig/epoch-499.pt
new file mode 100644
index 0000000000000000000000000000000000000000..8b85daaaa8ef8995c2c8c0fa208c89a385409fb3
--- /dev/null
+++ b/run_vrp20_tiny_orig/epoch-499.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e8301b31562cc54ed63557ada9bed70b313e52fc0a58620b6246af7cda3c47e6
+size 8383429