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.

Sign up for GitHub

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

Jump to bottom

Mixed precision #37

Merged
merged 16 commits into from
Nov 17, 2023
Merged

Mixed precision #37

merged 16 commits into from
Nov 17, 2023

Commits on Aug 25, 2023

  1. Quantization test

    @nikiniki1
    nikiniki1 committed Aug 25, 2023
    Configuration menu
    Copy the full SHA
    97f2bdc View commit details
    Browse the repository at this point in the history

Commits on Oct 18, 2023

  1. torch amp

    @nikiniki1
    nikiniki1 committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    bf59bd3 View commit details
    Browse the repository at this point in the history

  2. Update .gitignore

    @nikiniki1
    nikiniki1 committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    28bd31d View commit details
    Browse the repository at this point in the history

  3. Quantization test

    @nikiniki1
    nikiniki1 committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    9a4a219 View commit details
    Browse the repository at this point in the history

  4. Update .gitignore

    @nikiniki1
    nikiniki1 committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    3372b9f View commit details
    Browse the repository at this point in the history

  5. amp update

    @nikiniki1
    nikiniki1 committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    d7392e9 View commit details
    Browse the repository at this point in the history

  6. Merge branch 'quantization' of https://github.com/ITMO-NSS-team/torch… 

    …_DE_solver into quantization
    @nikiniki1
    nikiniki1 committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    769801d View commit details
    Browse the repository at this point in the history

Commits on Oct 20, 2023

  1. amp update v3 

    Due to lbfgs optim it is required to change the logic of closure.
    @nikiniki1
    nikiniki1 committed Oct 20, 2023
    Configuration menu
    Copy the full SHA
    067956b View commit details
    Browse the repository at this point in the history

  2. Delete AMP.ipynb

    @nikiniki1
    nikiniki1 committed Oct 20, 2023
    Configuration menu
    Copy the full SHA
    5f7d3a9 View commit details
    Browse the repository at this point in the history

Commits on Oct 25, 2023

  1. Small fixes

    @nikiniki1
    nikiniki1 committed Oct 25, 2023
    Configuration menu
    Copy the full SHA
    b20f342 View commit details
    Browse the repository at this point in the history

Commits on Nov 2, 2023

  1. Updated cache with AMP 

    * Now cache is working with AMP
* Slight refactor cache module
    @nikiniki1
    nikiniki1 committed Nov 2, 2023
    Configuration menu
    Copy the full SHA
    cb95164 View commit details
    Browse the repository at this point in the history

  2. Update example_wave_physics.py

    @nikiniki1
    nikiniki1 committed Nov 2, 2023
    Configuration menu
    Copy the full SHA
    47879c4 View commit details
    Browse the repository at this point in the history

  3. Update example_wave_mixed_precision.py

    @nikiniki1
    nikiniki1 committed Nov 2, 2023
    Configuration menu
    Copy the full SHA
    a6194f3 View commit details
    Browse the repository at this point in the history

  4. Update setup.py

    @nikiniki1
    nikiniki1 committed Nov 2, 2023
    Configuration menu
    Copy the full SHA
    0bf1f45 View commit details
    Browse the repository at this point in the history

Commits on Nov 15, 2023

  1. Small fixes 

    Removed scaler state dict in cache module.
    @nikiniki1
    nikiniki1 committed Nov 15, 2023
    Configuration menu
    Copy the full SHA
    d11dff8 View commit details
    Browse the repository at this point in the history

Commits on Nov 17, 2023

  1. Cache fix for mat

    @nikiniki1
    nikiniki1 committed Nov 17, 2023
    Configuration menu
    Copy the full SHA
    19e27f4 View commit details
    Browse the repository at this point in the history