File size: 862 Bytes
002bd9b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
python -m debugpy --wait-for-client --listen 0.0.0.0:5678 \
-m src.train \
train_data='[coco-instance-local]' eval_data='[coco-instance-local]' \
+model=base_sca_multitask_v2 \
model.cache_dir=.model.cache/ \
training.do_train=True \
training.do_eval=True \
training.fp16=True \
training.num_masks_per_sample=16 \
training.per_device_train_batch_size=1 \
training.dataloader_num_workers=4 \
training.max_steps=99 \
training.logging_first_step=True \
training.logging_steps=5 \
training.evaluate_before_train=True \
training.max_eval_samples=3 \
training.eval_steps=50 \
training.save_steps=50 \
wandb.log=False \
training.lr_scheduler_type=cosine \
+data_transforms=lsj-0_1-2_0 \
model.lm_head_model_name_or_path=gpt2 \
model.sam_model_name_or_path=facebook/sam-vit-base
|