jasonyma commited on
Commit
6812095
1 Parent(s): b2a2267

Upload config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +16 -0
config.yaml ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ device: cuda
3
+ agent:
4
+ _target_: liv.LIV
5
+ scratch: false
6
+ grad_text: true
7
+ metric: cos
8
+ device: ${device}
9
+ lr: 1.0e-5
10
+ weight_decay: 0.001
11
+ modelid: RN50
12
+ clipweight: 1.0
13
+ visionweight: 1.0
14
+ langweight: 0.0
15
+ gamma: 0.98
16
+ num_negatives: 0