Upload 3 files
Browse files- configs/base_checkpoint.pt +3 -0
- configs/example.yaml +20 -0
- configs/test.csv +0 -0
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
|
|
|