@@ -29,35 +29,30 @@ O código desenvolvido para o processo de treinamento e validação pode ser reu
29
29
30
30
Os seguintes parâmetros devem ser modificados/atualizados/conferidos antes de inicar um novo treinamento:
31
31
32
- REGION= #região em que a instância de processamento está lcoalizada no GCP (string)
33
-
34
- MODEL_NAME= #nome arbitrário do modelo (string)
35
-
36
- EPOCHS= #quantidade de épocas de treinamento (int)
37
-
38
- BATCH_SIZE= #tamanho do batch de treinamento (int)
39
-
40
- LAYERS= #camadas completamente conectadas a serem construídas no modelo (json)
41
- exemplo: "[ {\" class_name\" :\" Dense\" ,\" params\" :{\" units\" :256,\" activation\" :\" sigmoid\" }},{\" class_name\" :\" Dropout\" ,\" params\" :{\" rate\" :0.5}}] "
42
-
43
- BASE_PATH= #endereço de localização do dataset em tfrecords
44
-
45
- BUCKET_NAME= #nome do bucket no GCP
32
+ REGION= #região em que a instância de processamento está lcoalizada no GCP (string)
33
+ MODEL_NAME= #nome arbitrário do modelo (string)
34
+ EPOCHS= #quantidade de épocas de treinamento (int)
35
+ BATCH_SIZE= #tamanho do batch de treinamento (int)
36
+ LAYERS= #camadas completamente conectadas a serem construídas no modelo (json)
37
+ exemplo: "[{\"class_name\":\"Dense\",\"params\":{\"units\":256,\"activation\":\"sigmoid\"}},{\"class_name\":\"Dropout\",\"params\":{\"rate\":0.5}}]"
38
+ BASE_PATH= #endereço de localização do dataset em tfrecords
39
+ BUCKET_NAME= #nome do bucket no GCP
46
40
47
41
O comando a seguir inicia o treinamento utilizando as variáveis definidas acima, e outras que também devem ser atualizadas:
48
- gcloud ml-engine jobs submit training ${JOB_NAME} \
49
- --scale-tier=BASIC \
50
- --job-dir=${JOB_DIR} \
51
- --runtime-version=1.10 \
52
- --region=${REGION} \
53
- --module-name=image-classifier.train \
54
- --package-path=image-classifier \
55
- -- \
56
- --bucket-name=${BUCKET_NAME} \
57
- --epochs=${EPOCHS} \
58
- --batch-size=${BATCH_SIZE} \
59
- --current-date=${CURRENT_DATE} \
60
- --base-path=${BASE_PATH} \
61
- --loss=categorical_hinge \
62
- --layers=$LAYERS \
63
- --activation=softmax
42
+
43
+ gcloud ml-engine jobs submit training ${JOB_NAME} \
44
+ --scale-tier=BASIC \
45
+ --job-dir=${JOB_DIR} \
46
+ --runtime-version=1.10 \
47
+ --region=${REGION} \
48
+ --module-name=image-classifier.train \
49
+ --package-path=image-classifier \
50
+ -- \
51
+ --bucket-name=${BUCKET_NAME} \
52
+ --epochs=${EPOCHS} \
53
+ --batch-size=${BATCH_SIZE} \
54
+ --current-date=${CURRENT_DATE} \
55
+ --base-path=${BASE_PATH} \
56
+ --loss=categorical_hinge \
57
+ --layers=$LAYERS \
58
+ --activation=softmax
0 commit comments