1.确保您已经安装了Google Cloud SDK,并且您的环境变量已经正确配置,以便您可以从终端运行gcloud命令。
2.在终端中运行以下命令以创建一个Google Cloud项目,并确保您的gcloud命令行工具已切换到该项目:
gcloud projects create PROJECT-ID
gcloud config set project PROJECT-ID
3.确保您已经启用Vertex AI服务,您可以通过运行以下命令来检查:
gcloud services list --available
如果您发现Vertex AI服务未启用,则可以使用以下命令启用:
gcloud services enable aiplatform.googleapis.com
4.创建Hypertune作业的yaml配置文件。例如,您可以使用以下文件:
# hypertune.yaml
algorithm: random_sample
parameters:
- parameterName: lr
type: double
minValue: 0.0001
maxValue: 0.1
scaleType: unit_linear_scale
- parameterName: momentum
type: double
minValue: 0.1
maxValue: 0.9
scaleType: unit_linear_scale
此文件将配置一个随机采样算法,并定义两个超参数:lr和momentum。您可以根据您的实际需要自定义此yaml配置文件。
5.使用以下命令提交Hypertune作业:
gcloud ai custom-jobs submit hypertune \
--region=REGION \
--config=hypertune.yaml \
--display-name=DISPLAY-NAME \
--trial-image-uri=TRIAL-IMAGE-URI \
--objective-metric-name=OBJECTIVE-METRIC-NAME \
--objective-metric-minimize \
--parameter-server-type=CPU \
--max-trial-count=MAX-TRIAL-COUNT \
--max-concurrent-trial-count=MAX-CONCURRENT-TRIAL-COUNT \
--hyperparameter=TrialParameters.json \
--use-chief-in-tf-config