Update main.py

RubensZimbres · Mar 7, 2021 · 274b4d0 · 274b4d0
1 parent e998c46
commit 274b4d0
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Google-Cloud-Functions-Job/main.py b/Google-Cloud-Functions-Job/main.py
@@ -2,19 +2,19 @@
 from googleapiclient import discovery
 from oauth2client.client import GoogleCredentials
 
-id = 'machinelearning-XXXX'
-bucket_name = "bucket-XXX"
+id = 'machinelearning-XXX'
+bucket_name = "keras-XXX-folder"
 project_id = 'projects/{}'.format(id)
 job_name = "training_" + datetime.datetime.now().strftime("%y%m%d_%H%M%S")
 
 def main(event, context):
 
      training_inputs = {
      'scaleTier': 'BASIC',
-     'packageUris': [f"gs://{bucket_name}/package/trainer-0.1.tar.gz"],
+     'packageUris': ["gs://{}/trainer".format(bucket_name)],
      'pythonModule': 'trainer.task',
-     'region': 'asia-northeast1',
-     'jobDir': f"gs://{bucket_name}",
+     'region': 'us-central1',
+     'jobDir': "gs://{}".format(bucket_name),
      'runtimeVersion': '2.2',
      'pythonVersion': '3.7',
           }