can the author provide whole finetuning code?
i see the finetune description as:
train_command = [
'python', '-m', 'training.main',
'--name', name,
'--save-frequency', '5',
'--zeroshot-frequency', '10',
'--report-to', 'wandb',
'--train-data', train_csv,
'--csv-img-key', 'img_path',
'--csv-caption-key', 'label',
'--warmup', '10',
'--batch-size', '64',
'--lr', '5e-6',
'--wd', '0.1',
'--epochs', '10',
'--workers', '16',
'--model', model_name,
'--csv-separator', ',',
'--pretrained', pretrained_weight_path,
'--lock-text-freeze-layer-norm',
'--lock-image-freeze-bn-stats',
'--coca-caption-loss-weight','0',
'--coca-contrastive-loss-weight','1',
'--val-frequency', '10',
'--aug-cfg', 'color_jitter=(0.32, 0.32, 0.32, 0.08)', 'color_jitter_prob=0.5', 'gray_scale_prob=0'
]
subprocess.run(train_command)
but i can't find the "training.main" file , thus i can't do the finetune work
I sincerely want the author provide the whole finetune code Thanks~
Excuse me，have you received the fine-tuned model code from the author?
Excuse me，have you received the fine-tuned model code from the author?
no . The author told me that in github:
Hi biofocker,
Thanks for your interest in our work!
Our fine-tuning process leverages the training scaffolding from the excellent OpenCLIP repository. Please head over to the OpenCLIP GitHub for the complete training scripts, configuration files, and step-by-step instructions.
Let us know if you have other questions.
Best,
Weiqing
Thanks for your reply. Wishing you continued success in your work!