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。
wandb
如果你用的是运行Python 2的Estimators，你需要在安装wandb之前直接从一个轮子（wheel）来安装 psutil
https://wheels.galaxyproject.org/packages/psutil-5.4.8-cp27-cp27mu-manylinux1_x86_64.whl
wandb
​Github上有一个完整示例，还可到我们的论坛详细了解。

最近更新 1yr ago