Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

pad input to use tensor core #4911

Merged
merged 1 commit into from
Oct 22, 2020

Conversation

zhangting2020
Copy link
Contributor

@zhangting2020 zhangting2020 commented Oct 20, 2020

在开启dali的同时,输入shape修改为[4, 224, 224],会自动对通道进行pad。
4通道下,第一层卷积能够利用tensor core加速,模型提速6.4%

@CLAassistant
Copy link

CLAassistant commented Oct 20, 2020

CLA assistant check
All committers have signed the CLA.

Copy link
Contributor

@Xreki Xreki left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@zhangting2020 zhangting2020 merged commit 2392894 into PaddlePaddle:develop Oct 22, 2020
@zhangting2020 zhangting2020 deleted the pad_out branch October 22, 2020 10:53
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.

None yet

3 participants