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 及相关测试

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.

1 participant