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