#!/bin/bash

export HIP_VISIBLE_DEVICES=1

torchrun --nproc_per_node 4 -m open_clip_train.main \
--train-data '/home/datasets/clip/mscoco/{00000..00059}.tar' \
--train-num-samples 100000 \
--dataset-type webdataset \
--batch-size 320 \
--precision amp_bf16 \
--workers 4 \
--imagenet-val "/home/datasets/imagenet2012/val"