PICS / scripts /train.sh
Hang Zhou
Upload folder using huggingface_hub
9aff0cd verified
raw
history blame contribute delete
270 Bytes
# train on the whole dataset
python run_train.py \
--root_dir 'LOGS/all_data' \
--batch_size 16 \
--logger_freq 1000 \
--is_joint
python run_train.py \
--root_dir 'LOGS/lvis' \
--batch_size 16 \
--logger_freq 1000 \
--dataset_name lvis