Elfenreigen commited on
Commit
e695911
·
verified ·
1 Parent(s): 32bd57e

Upload 3 files

Browse files
configs/base_checkpoint.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13da6980aeda9569973b15efa0c5ed5c98f9c768859cd18fc024bf52b01cfb75
3
+ size 1095746872
configs/example.yaml ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ test_file: './configs/test.csv'
2
+ # test file: image_path, label
3
+
4
+ image_res: 512
5
+ batch_size: 64
6
+ test_batch_size: 128
7
+ num_workers: 8
8
+ test_num_workers: 8
9
+
10
+ temp: 0.07
11
+ mlm_probability: 0.15
12
+ queue_size: 8192
13
+ momentum: 0.995
14
+ alpha: 0.4
15
+ lam: [1.0,0.4]
16
+ image_encoder_name: resnet152
17
+ loss_ratio: 1.0
18
+ loss_ratio_note: clip_ratio
19
+ finetune: './configs/base_checkpoint.pt'
20
+ # inference ckpt location
configs/test.csv ADDED
The diff for this file is too large to render. See raw diff