Skip to content
This repository was archived by the owner on Apr 17, 2023. It is now read-only.

Conversation

@daniil-lyakhov
Copy link

@daniil-lyakhov daniil-lyakhov commented Oct 12, 2022

Purpose of this PR is to check NNCF is working in FP16 precision. This is basic POC which could possibly broke other scenarios.

PR introduces Fp16OptimizerHook usage for gradient scaling during training loop. With this code and updated version NNCF (with FP16 kernel improvements openvinotoolkit/nncf#1279) It's possible to optimize HRnet large-mod3 in FP16 precision

Copy link

@alexsu52 alexsu52 left a comment

Choose a reason for hiding this comment

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

Should NNCF version be updated for this changes?

@daniil-lyakhov
Copy link
Author

Should NNCF version be updated for this changes?

Yes, it should

@daniil-lyakhov
Copy link
Author

Purpose of this PR is to check NNCF is working in FP16 precision. This is basic POC which could possibly broke other scenarios.

@alexsu52
Copy link

Should NNCF version be updated for this changes?

Yes, it should

Let's update it here

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants