Skip to content

【训练营】小模型训练支持 - #219

Open
lewis56654281 wants to merge 1 commit into
InfiniTensor:masterfrom
lewis56654281:ztc/mnist-cnn-support
Open

lewis56654281 wants to merge 1 commit into
InfiniTensor:masterfrom
lewis56654281:ztc/mnist-cnn-support

Conversation

@lewis56654281

Copy link
Copy Markdown

为 MNIST 示例增加两层 CNN,支持 CPU、CUDA 和双卡 DDP 训练,补齐 Conv2d、ReLU 及相关测试

@kilinchange

Copy link
Copy Markdown
Collaborator

请解决 pr 冲突。

@kilinchange
kilinchange self-requested a review September 23, 2026 08:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants