mlfoundations / open_clip

An open source implementation of CLIP.
Other
9.93k stars 959 forks source link

How to save checkpoints? #754

Closed jyC23333 closed 10 months ago

jyC23333 commented 10 months ago

When I finish this command, where to find the training checkpoints?

python -m training.main \
    --save-frequency 1 \
    --zeroshot-frequency 1 \
    --report-to tensorboard \
    --train-data="/path/to/train_data.csv"  \
    --val-data="/path/to/validation_data.csv"  \
    --csv-img-key filepath \
    --csv-caption-key title \
    --imagenet-val=/path/to/imagenet/root/val/ \
    --warmup 10000 \
    --batch-size=128 \
    --lr=1e-3 \
    --wd=0.1 \
    --epochs=30 \
    --workers=8 \
    --model RN50
gabrielilharco commented 10 months ago

Hi @jyC23333. Training artifacts are logged at --logs, which defaults to ./logs/.

jyC23333 commented 10 months ago

@gabrielilharco Thanks, I found it.