accum_freq: 1 audio_ext: flac aug_cfg: {} batch_size: 64 beta1: 0.9 beta2: 0.95 cache_dir: None caption_weight: 0.0 checkpoint_path: U2_audio_scratch_locktext/checkpoints classification_label_key: labels classification_num_classes: 527 classification_weight: 0.0 coca_caption_loss_weight: 2.0 coca_contrastive_loss_weight: 1.0 compile_mode: default contrastive_weight: 1.0 copy_codebase: False csv_caption_key: title csv_img_key: filepath csv_separator: dac_model_type: 44khz dac_num_codebooks: 1 dac_tokenize: False data_filling: pad data_truncating: rand_trunc dataset_resampled: True dataset_type: webdataset-audio ddp_static_graph: True debug: False delete_previous_checkpoint: False device: cuda:0 dist_backend: None dist_url: None distill: False distill_model: None distill_pretrained: None distributed: True enable_fusion: False epochs: 5 epochs_cooldown: None eps: 1e-08 force_context_length: None force_custom_text: False force_image_size: None force_patch_dropout: None force_quick_gelu: False fsdp2: False gather_with_grad: True grad_checkpointing: False grad_clip_norm: 1.0 horovod: False image_interpolation: None image_mean: None image_resize_mode: None image_std: None imagenet_v2: None imagenet_val: None int16_normalize: False local_loss: True local_rank: 0 lock_audio: False lock_image: False lock_image_freeze_bn_stats: False lock_image_unlocked_groups: 0 lock_text: True lock_text_freeze_layer_norm: False lock_text_unlocked_layers: 0 log_every_n_steps: 100 log_level: 20 log_local: False log_path: U2_audio_scratch_locktext/out.log log_step_timings: False logs: voiceclap/checkpoints loss_dist_impl: None lr: 0.0003 lr_cooldown_end: 0.0 lr_cooldown_power: 1.0 lr_scheduler: cosine model: CLAPNaFlex-base-MiniLM momentum: None naflex: True name: U2_audio_scratch_locktext no_set_device_rank: False opt: adamw pack_max_clips: 4 precision: amp_bfloat16 pretrained: pretrained_audio: None pretrained_image: False rank: 0 remote_sync: None remote_sync_frequency: 300 remote_sync_protocol: s3 report_to: tensorboard resume: None save_frequency: 1 save_most_recent: True seed: 0 sequence_packing: False siglip: True skip_scheduler: False synthetic_seq_len: 0 tensorboard: True tensorboard_path: U2_audio_scratch_locktext/tensorboard torchcompile: False torchscript: False trace: False train_data: laions_got_talent_clean_with_captions/{000000..000823}.tar::emolia-balanced-5M-subset-flac48k-wds/emolia-{000000..000263}.tar::emolia-balanced-5M-subset-flac48k-wds/emolia-000265.tar::majestrino-data/{000000..000474}.tar::synthetic_vocal_bursts/{000000..000158}.tar::ears/{000000..000008}.tar::improved_synthetic_vocal_burts/{000000..000007}.tar train_data_file: None train_data_upsampling_factors: None train_num_samples: 4000000 use_bn_sync: False use_bnb_linear: None val_data: None val_frequency: 1 val_num_samples: None wandb: False wandb_notes: wandb_project_name: open-clip warmup: 1000 wd: 0.2 workers: 12 world_size: 16 zeroshot_frequency: 1