MIL-NCE_HowTo100M:适用于MIL-NCE HowTo100M的PyTorch GPU分布式培训代码

上传者: 42131628 | 上传时间: 2022-05-16 17:08:20 | 文件大小: 22.02MB | 文件类型: ZIP
使用PyTorch在GPU上进行MIL-NCE端到端HowTo100M培训 此存储库包含CVPR'20论文的开源PyTorch分布式培训代码:[1]。 [1]中的原始代码库依赖于Google和DeepMind的内部工具以及TPU v3加速器的使用,这使其难以按原样发布。 相反,此存储库使用PyTorch / ffmpeg和合理数量的GPU提供了[1]的实现。 培训代码在法国公共AI集群(请参阅下面的致谢)。 它经过专门设计,可在基于SLURM的集群管理上运行,以进行多节点分布式培训,但可以轻松地针对任何其他集群管理系统进行修改。 本文的开源PyTorch实现有一些细微的差异,例如: 使用余弦学习速率衰减代替[1]中描述的逐步衰减。 没有在不同的GPU和节点之间共享批处理规范化统计信息,因为在GPU上执行此类操作比TPU慢得多。 使用略微不同的时空训练视频分辨率的输入视频剪辑。

文件下载

资源详情

[{"title":"( 22 个子文件 22.02MB ) MIL-NCE_HowTo100M:适用于MIL-NCE HowTo100M的PyTorch GPU分布式培训代码","children":[{"title":"MIL-NCE_HowTo100M-master","children":[{"title":"log","children":[{"title":"readme.txt <span style='color:#111;'> 47B </span>","children":null,"spread":false}],"spread":true},{"title":"msrvtt_loader.py <span style='color:#111;'> 4.26KB </span>","children":null,"spread":false},{"title":"utils.py <span style='color:#111;'> 1.40KB </span>","children":null,"spread":false},{"title":"eval_youcook.py <span style='color:#111;'> 2.75KB </span>","children":null,"spread":false},{"title":"s3dg.py <span style='color:#111;'> 13.21KB </span>","children":null,"spread":false},{"title":"eval_msrvtt.py <span style='color:#111;'> 2.74KB </span>","children":null,"spread":false},{"title":"loss.py <span style='color:#111;'> 616B </span>","children":null,"spread":false},{"title":"hmdb_loader.py <span style='color:#111;'> 3.13KB </span>","children":null,"spread":false},{"title":"LICENSE <span style='color:#111;'> 11.09KB </span>","children":null,"spread":false},{"title":"metrics.py <span style='color:#111;'> 798B </span>","children":null,"spread":false},{"title":"video_loader.py <span style='color:#111;'> 5.69KB </span>","children":null,"spread":false},{"title":"README.md <span style='color:#111;'> 13.51KB </span>","children":null,"spread":false},{"title":"eval_hmdb.py <span style='color:#111;'> 3.86KB </span>","children":null,"spread":false},{"title":"csv","children":[{"title":"msrvtt_test.csv <span style='color:#111;'> 72.73KB </span>","children":null,"spread":false},{"title":"howto100m_videos.csv <span style='color:#111;'> 18.97MB </span>","children":null,"spread":false},{"title":"validation_youcook.csv <span style='color:#111;'> 222.60KB </span>","children":null,"spread":false},{"title":"all_videos.csv <span style='color:#111;'> 18.97MB </span>","children":null,"spread":false},{"title":"hmdb51.csv <span style='color:#111;'> 481.74KB </span>","children":null,"spread":false}],"spread":false},{"title":"args.py <span style='color:#111;'> 5.79KB </span>","children":null,"spread":false},{"title":"main_distributed.py <span style='color:#111;'> 10.72KB </span>","children":null,"spread":false},{"title":"youcook_loader.py <span style='color:#111;'> 4.78KB </span>","children":null,"spread":false},{"title":"checkpoint","children":[{"title":"readme.txt <span style='color:#111;'> 47B </span>","children":null,"spread":false}],"spread":false}],"spread":false}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明