Chinese-CLIP icon indicating copy to clipboard operation
Chinese-CLIP copied to clipboard

finetune时的问题

Open DengyingFu opened this issue 11 months ago • 2 comments

image 请问这是什么原因 我的配置如下 image

DengyingFu avatar Mar 13 '24 09:03 DengyingFu

同遇到过,单机单卡跑的话,可以用下面的命令,感觉是pytorch版本或者参数的问题 `

单机单卡用下面的命令,不要加分布式参数:

python cn_clip/training/main.py
--train-data=${train_data}
--val-data=${val_data}
--resume=${resume}
--logs=${output_base_dir}
--name=${name}
--save-step-frequency=${save_step_frequency}
--save-epoch-frequency=${save_epoch_frequency}
--log-interval=${log_interval}
${report_training_batch_acc}
--context-length=${context_length}
--warmup=${warmup}
--batch-size=${batch_size}
--valid-batch-size=${valid_batch_size}
--valid-step-interval=${valid_step_interval}
--valid-epoch-interval=${valid_epoch_interval}
--accum-freq=${accum_freq}
--lr=${lr}
--wd=${wd}
--max-epochs=${max_epochs}
--vision-model=${vision_model}
${use_augment}
--text-model=${text_model} `

ChesonHuang avatar Apr 10 '24 12:04 ChesonHuang

export MASTER_ADDR=localhost 我改了这个就可以啦

xiaoAugenstern avatar Sep 11 '24 08:09 xiaoAugenstern