models/scripts/test_elasticdl_submit.sh at develop · sql-machine-learning/models · GitHub

23 lines (22 loc) · 912 Bytes

#!/bin/bash

elasticdl train --image_base=sqlflow/sqlflow_models \

--model_def=dnnclassifier.DNNClassifier \

--training_data=sqlflow_test_iris_train \

--data_reader_params='columns=["sepal_length", "sepal_width", "petal_length", "petal_width", "class"];label_col="class"' \

--envs="ODPS_PROJECT_NAME=gomaxcompute_driver_w7u,ODPS_ACCESS_ID=$ODPS_ACCESS_ID,ODPS_ACCESS_KEY=$ODPS_ACCESS_KEY" \

--minibatch_size=32 \

--num_epochs=2 \

--model_zoo=/sqlflow_models \

--job_name=test-odps \

--num_minibatches_per_task=2 \

--image_pull_policy=Never \

--num_workers=2 \

--num_ps_pods=1 \

--master_resource_request="cpu=200m,memory=128Mi" \

--master_resource_limit="cpu=1,memory=2048Mi" \

--worker_resource_request="cpu=200m,memory=128Mi" \

--worker_resource_limit="cpu=1,memory=3072Mi" \

--ps_resource_request="cpu=200m,memory=128Mi" \

--ps_resource_limit="cpu=1,memory=2048Mi" \

--grads_to_wait=2 \

--output=model_output

Expand file tree