Skip to content

Commit 52c10a8

Browse files
vrkharenadiaya
authored andcommitted
Update train-coach.py (aws#505)
updating comments in the file
1 parent d02f3eb commit 52c10a8

File tree

1 file changed

+2
-3
lines changed
  • reinforcement_learning/rl_predictive_autoscaling_coach_customEnv/src

1 file changed

+2
-3
lines changed

reinforcement_learning/rl_predictive_autoscaling_coach_customEnv/src/train-coach.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,9 @@ def _save_tf_model(self):
5151
export_dir='model',
5252
inputs={"observation": input_nodes},
5353
outputs={"policy": output_nodes.outputs[0]})
54-
# Move to the appropriate folder. Don't mind the directory, this just works.
55-
# rl-cart-pole is the name of the model. Remember it.
54+
# Move to the appropriate folder.
5655
shutil.move('model/', model_dir + '/model/tf-model/00000001/')
57-
# EASE will pick it up and upload to the right path.
56+
# SageMaker will pick it up and upload to the right path.
5857
print("Success")
5958

6059

0 commit comments

Comments
 (0)