加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
distributed_train.sh 107 Bytes
一键复制 编辑 原始数据 按行查看 历史
Ross Wightman 提交于 2019-04-05 20:49 . Fix distributed train script
#!/bin/bash
NUM_PROC=$1
shift
python -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@"
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化