Spaces:
Running
on
Zero
Running
on
Zero
| # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved | |
| BIN="python tools/train_net.py" | |
| OUTPUT="inference_test_output" | |
| NUM_GPUS=2 | |
| CFG_LIST=( "${@:1}" ) | |
| if [ ${#CFG_LIST[@]} -eq 0 ]; then | |
| CFG_LIST=( ./configs/quick_schedules/*inference_acc_test.yaml ) | |
| fi | |
| echo "========================================================================" | |
| echo "Configs to run:" | |
| echo "${CFG_LIST[@]}" | |
| echo "========================================================================" | |
| for cfg in "${CFG_LIST[@]}"; do | |
| echo "========================================================================" | |
| echo "Running $cfg ..." | |
| echo "========================================================================" | |
| $BIN \ | |
| --eval-only \ | |
| --num-gpus $NUM_GPUS \ | |
| --config-file "$cfg" \ | |
| OUTPUT_DIR $OUTPUT | |
| rm -rf $OUTPUT | |
| done | |
| echo "========================================================================" | |
| echo "Running demo.py ..." | |
| echo "========================================================================" | |
| DEMO_BIN="python demo/demo.py" | |
| COCO_DIR=datasets/coco/val2014 | |
| mkdir -pv $OUTPUT | |
| set -v | |
| $DEMO_BIN --config-file ./configs/quick_schedules/panoptic_fpn_R_50_inference_acc_test.yaml \ | |
| --input $COCO_DIR/COCO_val2014_0000001933* --output $OUTPUT | |
| rm -rf $OUTPUT | |