FluxML/Flux.jl

Allow BatchNorm training on CUDA with `track_stats=false`

已关闭

#1,606 创建于 2021年5月31日

在 GitHub 查看贡献者指南
 (4 条评论) (0 个反应) (0 位负责人)Julia (619 个派生)batch import
good first issuehelp wanted

仓库指标

星标
 (4,725 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

Gathered from https://discourse.julialang.org/t/batchnorm-only-track-stats-true-supported-on-gpu/62091.

This would most likely require changes in NNlibCUDA as well. I'm not sure how interchangeable the various cudnnBatchNormalizationForward* functions are, so putting a pin in this until someone more knowledgeable can comment.

贡献者指南

研究方向
检查Flux.jl和NNlibCUDA.jl中的Batchnorm实现。了解cudnnBatchNormalizationForward函数并修改它们以支持track stats=false。在支持CUDA的GPU上测试更改。
技术栈
无
领域
backendmachine learning
议题类型
功能
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
3
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
3-5 天
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
基本清晰
前置要求
JuliaCUDAGit
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
30