params/2022.yaml

accum_freq: 1
aug_cfg: {}
batch_size: 704
beta1: 0.9
beta2: 0.98
checkpoint_path: /tmp/restart_2022/datacomp_xlarge-basic_cumulative_seq_2022/checkpoints
coca_caption_loss_weight: 2.0
coca_contrastive_loss_weight: 1.0
copy_codebase: False
csv_caption_key: title
csv_img_key: filepath
csv_separator: 	
dataset_resampled: True
dataset_type: webdataset
ddp_static_graph: True
debug: False
decay_fraction: 0.2
delete_previous_checkpoint: False
device: cuda:0
dist_backend: nccl
dist_url: env://
distill: False
distill_model: None
distill_pretrained: None
distributed: True
epochs: 32
epochs_cooldown: None
eps: 1e-06
force_custom_text: False
force_image_size: None
force_patch_dropout: None
force_qk_norm: False
force_qk_norm_eps: 1e-05
force_quick_gelu: False
gather_with_grad: True
grad_checkpointing: True
grad_clip_norm: None
horovod: False
image_mean: None
image_std: None
imagenet_v2: None
imagenet_val: ../imagenet_validation
is_iteration_based: True
local_loss: True
local_rank: 0
lock_image: False
lock_image_freeze_bn_stats: False
lock_image_unlocked_groups: 0
lock_text: False
lock_text_freeze_layer_norm: False
lock_text_unlocked_layers: 0
log_every_n_steps: 100
log_level: 20
log_local: False
log_path: /tmp/restart_2022/datacomp_xlarge-basic_cumulative_seq_2022/out.log
logs: /tmp/restart_2022
lr: 0.001
lr_cooldown_end: 0.0
lr_cooldown_power: 1.0
lr_scheduler: cosine
max_iterations: 20500
model: ViT-B-16
name: datacomp_xlarge-basic_cumulative_seq_2022
new_run: True
no_set_device_rank: False
precision: amp
pretrained: 
pretrained_image: False
rank: 0
remote_sync: xlarge_CL_basic_filter/restart_2022/
remote_sync_frequency: 300
remote_sync_protocol: s3
report_to: wandb
resume: xlarge_CL_basic_filter/restart_2021/datacomp_xlarge-basic_cumulative_seq_2021/checkpoints/epoch_4.pt
save_frequency: 1
save_most_recent: True
seed: 0
skip_scheduler: False
tensorboard: False
tensorboard_path: 
torchcompile: False
torchscript: False
trace: False
train_data: xlarge_basic_filter/2022/1/{00000000..00000639}.tar::xlarge_basic_filter/2022/10/{00000000..00000575}.tar::xlarge_basic_filter/2022/11/{00000000..00000639}.tar::xlarge_basic_filter/2022/12/{00000000..00000639}.tar::xlarge_basic_filter/2022/13/{00000000..00000703}.tar::xlarge_basic_filter/2022/14/{00000000..00000639}.tar::xlarge_basic_filter/2022/15/{00000000..00000639}.tar::xlarge_basic_filter/2022/16/{00000000..00000599}.tar::xlarge_basic_filter/2022/17/{00000000..00000571}.tar::xlarge_basic_filter/2022/18/{00000000..00000703}.tar::xlarge_basic_filter/2022/19/{00000000..00000639}.tar::xlarge_basic_filter/2022/2/{00000000..00000698}.tar::xlarge_basic_filter/2022/20/{00000000..00000599}.tar::xlarge_basic_filter/2022/21/{00000000..00000599}.tar::xlarge_basic_filter/2022/22/{00000000..00000559}.tar::xlarge_basic_filter/2022/23/{00000000..00000703}.tar::xlarge_basic_filter/2022/24/{00000000..00000447}.tar::xlarge_basic_filter/2022/25/{00000000..00000447}.tar::xlarge_basic_filter/2022/26/{00000000..00000063}.tar::xlarge_basic_filter/2022/27/{00000000..00000127}.tar::xlarge_basic_filter/2022/28/{00000000..00000767}.tar::xlarge_basic_filter/2022/29/{00000000..00000767}.tar::xlarge_basic_filter/2022/3/{00000000..00000698}.tar::xlarge_basic_filter/2022/30/{00000000..00000740}.tar::xlarge_basic_filter/2022/31/{00000000..00000703}.tar::xlarge_basic_filter/2022/32/{00000000..00000703}.tar::xlarge_basic_filter/2022/33/{00000000..00000732}.tar::xlarge_basic_filter/2022/34/{00000000..00000703}.tar::xlarge_basic_filter/2022/35/{00000000..00000713}.tar::xlarge_basic_filter/2022/36/{00000000..00000767}.tar::xlarge_basic_filter/2022/37/{00000000..00000767}.tar::xlarge_basic_filter/2022/38/{00000000..00000703}.tar::xlarge_basic_filter/2022/39/{00000000..00000640}.tar::xlarge_basic_filter/2022/4/{00000000..00000640}.tar::xlarge_basic_filter/2022/40/{00000000..00000639}.tar::xlarge_basic_filter/2022/41/{00000000..00000575}.tar::xlarge_basic_filter/2022/42/{00000000..00000649}.tar::xlarge_basic_filter/2022/43/{00000000..00000689}.tar::xlarge_basic_filter/2022/44/{00000000..00000703}.tar::xlarge_basic_filter/2022/45/{00000000..00000703}.tar::xlarge_basic_filter/2022/46/{00000000..00000679}.tar::xlarge_basic_filter/2022/47/{00000000..00000703}.tar::xlarge_basic_filter/2022/48/{00000000..00000767}.tar::xlarge_basic_filter/2022/49/{00000000..00000767}.tar::xlarge_basic_filter/2022/5/{00000000..00000703}.tar::xlarge_basic_filter/2022/50/{00000000..00000767}.tar::xlarge_basic_filter/2022/51/{00000000..00000767}.tar::xlarge_basic_filter/2022/52/{00000000..00000756}.tar::xlarge_basic_filter/2022/53/{00000000..00001159}.tar::xlarge_basic_filter/2022/54/{00000000..00001151}.tar::xlarge_basic_filter/2022/55/{00000000..00000359}.tar::xlarge_basic_filter/2022/56/{00000000..00000359}.tar::xlarge_basic_filter/2022/57/{00000000..00000383}.tar::xlarge_basic_filter/2022/58/{00000000..00000383}.tar::xlarge_basic_filter/2022/59/{00000000..00000359}.tar::xlarge_basic_filter/2022/6/{00000000..00000702}.tar::xlarge_basic_filter/2022/7/{00000000..00000703}.tar::xlarge_basic_filter/2022/8/{00000000..00000699}.tar::xlarge_basic_filter/2022/9/{00000000..00000617}.tar
train_data_upsampling_factors: None
train_num_samples: None
use_bn_sync: False
use_bnb_linear: None
val_data: None
val_frequency: 1
val_num_samples: None
wandb: True
wandb_notes: 
wandb_project_name: datacomp-xlarge-CL
warmup: 2000
wd: 0.2
workers: 4
world_size: 128
zeroshot_frequency: 1