do_train.sh 238 Bytes
Newer Older
Gunho Park's avatar
Gunho Park committed
1
2
3
4
#!/bin/bash
python3 train.py \
  --experiment=detr_coco \
  --mode=train_and_eval \
Gunho Park's avatar
Gunho Park committed
5
  --model_dir=gs://ghpark-ckpts/detr/detr_coco/ckpt_03_detr_coco_resnet101 \
Gunho Park's avatar
Gunho Park committed
6
7
  --tpu=postech-tpu \
  --params_override=runtime.distribution_strategy='tpu'