Elea Zhong commited on
Commit
3499dab
·
1 Parent(s): 1b9d6c7

add triplet losses conf

Browse files
configs/regression/triplet/mse-triplet-i.yaml ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ wandb_run_name: "reg-mse-triplet-f"
2
+ output_dir: "/data/checkpoints/reg-mse-triplet-f"
3
+
4
+ train_loss_terms:
5
+ mse: 1.0
6
+ triplet: 0.25
7
+
8
+ triplet_margin: 0.0
9
+ triplet_min_abs_diff: 0.25
configs/regression/triplet/mse-triplet-j.yaml ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ wandb_run_name: "reg-mse-triplet-f"
2
+ output_dir: "/data/checkpoints/reg-mse-triplet-f"
3
+
4
+ train_loss_terms:
5
+ mse: 1.0
6
+ triplet: 0.0625
7
+
8
+ triplet_margin: 0.0
9
+ triplet_min_abs_diff: 0.25
configs/regression/triplet/mse-triplet-k.yaml ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ wandb_run_name: "reg-mse-triplet-f"
2
+ output_dir: "/data/checkpoints/reg-mse-triplet-f"
3
+
4
+ train_loss_terms:
5
+ mse: 1.0
6
+ triplet: 0.015625
7
+
8
+ triplet_margin: 0.0
9
+ triplet_min_abs_diff: 0.25