Elea Zhong commited on
Commit
c4a6ce2
·
1 Parent(s): 6064267

add dist match yamls

Browse files
configs/regression/dm/mse-dm-a.yaml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ wandb_run_name: "reg-mse-dm-a"
2
+ output_dir: "/data/checkpoints/reg-mse-dm-a"
3
+
4
+ train_loss_terms:
5
+ mse: 1.0
6
+ distribution_matching: 1.0
configs/regression/dm/mse-dm-b.yaml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ wandb_run_name: "reg-mse-dm-b"
2
+ output_dir: "/data/checkpoints/reg-mse-dm-b"
3
+
4
+ train_loss_terms:
5
+ mse: 1.0
6
+ distribution_matching: 0.25
configs/regression/dm/mse-dm-c.yaml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ wandb_run_name: "reg-mse-dm-c"
2
+ output_dir: "/data/checkpoints/reg-mse-dm-c"
3
+
4
+ train_loss_terms:
5
+ mse: 1.0
6
+ distribution_matching: 0.0625
configs/regression/dm/mse-dm-d.yaml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ wandb_run_name: "reg-mse-dm-d"
2
+ output_dir: "/data/checkpoints/reg-mse-dm-d"
3
+
4
+ train_loss_terms:
5
+ mse: 1.0
6
+ distribution_matching: 0.01625