SageMaker的FensorFlow培训工作-如何更改脚本存档路径?

时间:2020-09-22 16:19:50

标签: python amazon-web-services tensorflow amazon-sagemaker

也许有人遇到过这样的问题:当由sagemaker.tensorflow.TensorFlow类在SageMaker中创建训练作业时,我需要设置一个参数,该参数负责将训练脚本代码保存在S3中,以使其路径成为精确: sagemaker_submit_directory module_dir -我不确定。默认情况下,它看起来像这样:

s3: //bucket/job-name/source/sourcedir.tar.gz. 

我需要在存储桶名称和作业名称之间添加一些内容。我该怎么做?我试图将此类参数添加到超参数中,但随后我的训练工作却被压垮了(不幸的是,由于堆栈跟踪大于限制,并且无法完全看到它,因此无法读取整个错误消息)?

有什么主意吗?

1 个答案:

答案 0 :(得分:0)

好的,这表明我在搜索文档时不够耐心。答案是参数“ code_location”:

https://sagemaker.readthedocs.io/en/stable/api/training/estimators.html?highlight=code_location#sagemaker.estimator.Framework