File size: 740 Bytes
7002f4e 2ca760b 7002f4e 2ca760b 7002f4e 2ca760b 7002f4e 2ca760b 7002f4e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | seed: 42
data:
root: data
protocol: remoteclip_rgb_bpe_pairs_v1
image_size: 224
channels: 3
context_length: 77
vocabulary_size: 49408
sot_token_id: 49406
eot_token_id: 49407
pad_token_id: 0
train_samples: 8
test_samples: 4
num_semantic_groups: 2
model:
image_size: 224
patch_size: 32
embed_dim: 64
vision_width: 64
vision_layers: 2
vision_heads: 4
text_width: 64
text_layers: 2
text_heads: 4
training:
epochs: 1
batch_size: 4
learning_rate: 0.0005
weight_decay: 0.01
num_workers: 0
runtime:
device: auto
amp: true
paths:
checkpoint: result/checkpoints/remoteclip.pt
training_metrics: result/training/metrics.json
inference_dir: result/output
evaluation_dir: result/evaluation
|