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

Mixed precision compound #159

Merged
merged 2 commits into from Mar 11, 2021
Merged

Mixed precision compound #159

merged 2 commits into from Mar 11, 2021

Conversation

maljoras
Copy link
Collaborator

@maljoras maljoras commented Mar 10, 2021

Related issues

Description

A new abstract device that implements an SGD optimizer that does not do the update step in fully analog mode, but instead computes the outer product in digital (assuming a high precision storage of the result) and then transfers it to the analog tile. For details, see Nandakumar et al. (2020) https://doi.org/10.3389/fnins.2020.00406

Details

To use this tile, one specifies it in the rpu_config, e.g.

rpu_config = DigitalRankUpdateRPUConfig(
device = MixedPrecisionCompound(device=ReRamESPresetDevice())

Presets are available as well. The mixed precision compound also supports quantization of activation and error as well as sequential transfer to the analog tile.

@diego-plan9 diego-plan9 merged commit 64e6850 into IBM:master Mar 11, 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.

None yet

2 participants