model/wallclock-regression-v3 (#104)

* updated weights for wallclock regression model (v3)
spacetelescope · Apr 16, 2021 · a543527 · a543527
1 parent d47a685
commit a543527
Show file tree

Hide file tree

Showing 8 changed files with 4 additions and 3 deletions.
diff --git a/lambda/JobPredict/models/wall_reg/3/saved_model.pb b/lambda/JobPredict/models/wall_reg/3/saved_model.pb
diff --git a/lambda/JobPredict/models/wall_reg/3/variables/variables.data-00000-of-00001 b/lambda/JobPredict/models/wall_reg/3/variables/variables.data-00000-of-00001
diff --git a/lambda/JobPredict/models/wall_reg/3/variables/variables.index b/lambda/JobPredict/models/wall_reg/3/variables/variables.index
diff --git a/lambda/JobPredict/models/wall_reg/3/weights/checkpoint b/lambda/JobPredict/models/wall_reg/3/weights/checkpoint
@@ -0,0 +1,2 @@
+model_checkpoint_path: "ckpt_3"
+all_model_checkpoint_paths: "ckpt_3"
diff --git a/lambda/JobPredict/models/wall_reg/3/weights/ckpt_3.data-00000-of-00001 b/lambda/JobPredict/models/wall_reg/3/weights/ckpt_3.data-00000-of-00001
diff --git a/lambda/JobPredict/models/wall_reg/3/weights/ckpt_3.index b/lambda/JobPredict/models/wall_reg/3/weights/ckpt_3.index
diff --git a/lambda/JobPredict/predict_handler.py b/lambda/JobPredict/predict_handler.py
@@ -139,7 +139,7 @@ def regressor(model, data):
 # load models
 clf = get_model("./models/mem_clf/2/")
 mem_reg = get_model("./models/mem_reg/2/")
-wall_reg = get_model("./models/wall_reg/2/")
+wall_reg = get_model("./models/wall_reg/3/")
 
 
 def lambda_handler(event, context):
@@ -158,7 +158,6 @@ def lambda_handler(event, context):
     MEMORY REGRESSION: A third regression model is used to estimate the actual value of memory needed for the job. This is mainly for the purpose of logging/future analysis and is not currently being used for allocating memory in calcloud jobs.
     """
     bucket_name = event["Bucket"]
-
     key = event["Key"]
     ipppssoot = event["Ipppssoot"]
     prep = Preprocess(ipppssoot, bucket_name, key)

diff --git a/terraform/lambda_job_predict.tf b/terraform/lambda_job_predict.tf
@@ -34,7 +34,7 @@ module "lambda_function_container_image" {
   package_type = "Image"
 
   timeout       = 360
-  memory_size   = 800
+  memory_size   = 1024
   cloudwatch_logs_retention_in_days = local.lambda_log_retention_in_days
 
   # ensures that terraform doesn't try to mess with IAM
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		model_checkpoint_path: "ckpt_3"
		all_model_checkpoint_paths: "ckpt_3"