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

Weight scaling #129

Merged
merged 3 commits into from
Feb 15, 2021
Merged

Weight scaling #129

merged 3 commits into from
Feb 15, 2021

Conversation

maljoras
Copy link
Collaborator

Related issues

Added feature.

Description

Here a conversion factor (weight scaling) is introduced that scales the weight range of the cross bar in respect to the digital output. For instance during initialization for training, the weight_scaling_omega factor can be used to set the absolute max of the weight matrix to some value (e.g. to the maximum of the weight range) and scale the output accordingly. This ensures that tile operates in a proper dynamic regime for a given DNN and usually improves achievable accuracy.

Details

See Rasch, Gokmen & Haensch (2019) for algorithmic details.

@maljoras maljoras requested a review from diego-plan9 February 12, 2021 17:15
Copy link
Member

@diego-plan9 diego-plan9 left a comment

Choose a reason for hiding this comment

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

Thanks Malte!

@diego-plan9 diego-plan9 merged commit aae7045 into IBM:master Feb 15, 2021
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