Documentation
搜索文档…
SageMaker

SageMaker集成

W&B与亚马逊SageMaker集成,自动地读取超参数、对分布式运行分组、从检查点开始断点续训。

认证

当wandb.init()被调用时,W&B会在训练脚本的相对路径下寻找一个secrets.env文件,并将其加载到环境中。你可以在你启动实验的脚本中,调用wandb.sagemaker_auth(path="source_dir"),来生成secrets.env文件。一定要把该文件添加到你的.gitignore
现有Estimators
如果你用的是SageMaker预配置的Estimators,你需要向源代码文件路径添加一个requirements.txt,其中包含wandb。
1
wandb
Copied!
如果你用的是运行Python 2的Estimators,你需要在安装wandb之前直接从一个轮子(wheel)来安装 psutil
1
https://wheels.galaxyproject.org/packages/psutil-5.4.8-cp27-cp27mu-manylinux1_x86_64.whl
2
wandb
Copied!
Github上有一个完整示例,还可到我们的论坛详细了解。
最近更新 9mo ago
复制链接