SageMaker
当wandb.init()被调用时,W&B会在训练脚本的相对路径下寻找一个
secrets.env
文件,并将其加载到环境中。你可以在你启动实验的脚本中,调用wandb.sagemaker_auth(path="source_dir")
,来生成secrets.env
文件。一定要把该文件添加到你的.gitignore
!现有Estimators
如果你用的是SageMaker预配置的Estimators,你需要向源代码文件路径添加一个
requirements.txt
,其中包含wandb。wandb
https://wheels.galaxyproject.org/packages/psutil-5.4.8-cp27-cp27mu-manylinux1_x86_64.whl
wandb
最近更新 1yr ago