d2bc3da0a1
* added UNet for medical image segmentation * added TF-AMP support for RN50 * small updates for other models (READMEs, benchmark & testing scripts)
11 lines
390 B
Bash
11 lines
390 B
Bash
PIPELINE_CONFIG_PATH=${1:-"/workdir/models/research/configs"}"/ssd320_full_1gpus.config"
|
|
|
|
export TF_ENABLE_AUTO_MIXED_PRECISION=1
|
|
|
|
SCRIPT_DIR=$(dirname "${BASH_SOURCE[0]}")
|
|
OBJECT_DETECTION=$(realpath $SCRIPT_DIR/../object_detection/)
|
|
PYTHONPATH=$PYTHONPATH:$OBJECT_DETECTION
|
|
|
|
python $SCRIPT_DIR/SSD320_inference.py \
|
|
--pipeline_config_path=${PIPELINE_CONFIG_PATH} \
|
|
"${@:2}"
|