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

Add option for weight tying on TPU's #5441

Merged
merged 32 commits into from
Feb 18, 2021
Merged

Add option for weight tying on TPU's #5441

merged 32 commits into from
Feb 18, 2021

Commits on Jan 19, 2021

  1. added on_post_move_to_device

    @lezwon
    lezwon committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    5cc9959 View commit details
    Browse the repository at this point in the history

  2. added tests

    @lezwon
    lezwon committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    3d1a313 View commit details
    Browse the repository at this point in the history

  3. docs and refactors

    @lezwon
    lezwon committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    bc7910b View commit details
    Browse the repository at this point in the history

  4. Update tests/backends/test_tpu_backend.py 

    Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
    @lezwon @Borda
    lezwon and Borda committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    bb4c891 View commit details
    Browse the repository at this point in the history

  5. Update docs/source/tpu.rst 

    Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
    @lezwon @Borda
    lezwon and Borda committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    9897945 View commit details
    Browse the repository at this point in the history

  6. Update docs/source/tpu.rst 

    Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
    @lezwon @Borda
    lezwon and Borda committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    0528db8 View commit details
    Browse the repository at this point in the history

  7. Update pytorch_lightning/core/decorators.py 

    Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
    @lezwon @Borda
    lezwon and Borda committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    c41e4ac View commit details
    Browse the repository at this point in the history

  8. Update pytorch_lightning/core/decorators.py 

    Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
    @lezwon @Borda
    lezwon and Borda committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    c7866ff View commit details
    Browse the repository at this point in the history

  9. Update docs/source/tpu.rst 

    Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
    @lezwon @rohitgr7
    lezwon and rohitgr7 committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    ac494d1 View commit details
    Browse the repository at this point in the history

  10. Update pytorch_lightning/core/decorators.py 

    Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
    @lezwon @rohitgr7
    lezwon and rohitgr7 committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    5e49706 View commit details
    Browse the repository at this point in the history

  11. Update pytorch_lightning/core/decorators.py 

    Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
    @lezwon @rohitgr7
    lezwon and rohitgr7 committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    e1126b2 View commit details
    Browse the repository at this point in the history

  12. Update pytorch_lightning/core/decorators.py 

    Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
    @lezwon @rohitgr7
    lezwon and rohitgr7 committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    975b899 View commit details
    Browse the repository at this point in the history

  13. Update pytorch_lightning/core/decorators.py 

    Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
    @lezwon @rohitgr7
    lezwon and rohitgr7 committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    814b163 View commit details
    Browse the repository at this point in the history

  14. Update pytorch_lightning/core/hooks.py 

    Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
    @lezwon @rohitgr7
    lezwon and rohitgr7 committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    e69910a View commit details
    Browse the repository at this point in the history

  15. moved weight sharing module back to test 

    updated tpu available
    @lezwon
    lezwon committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    8a22096 View commit details
    Browse the repository at this point in the history

  16. add count to warning

    @lezwon
    lezwon committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    bf5349b View commit details
    Browse the repository at this point in the history

  17. fix doctest

    @lezwon
    lezwon committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    35e67c0 View commit details
    Browse the repository at this point in the history

  18. import trainer in doctest

    @lezwon
    lezwon committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    829c041 View commit details
    Browse the repository at this point in the history

  19. import trainer in doctest

    @lezwon
    lezwon committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    a1e574f View commit details
    Browse the repository at this point in the history

  20. do not test code as no TPU device

    @lezwon
    lezwon committed Jan 19, 2021
    Configuration menu
    Copy the full SHA
    ac71a15 View commit details
    Browse the repository at this point in the history

  21. Merge branch 'release/1.2-dev' into bugfix/2705_weights_tying

    @lezwon
    lezwon authored Jan 19, 2021
    Configuration menu
    Copy the full SHA
    5a5f306 View commit details
    Browse the repository at this point in the history

Commits on Jan 20, 2021

  1. param count to layer count

    @lezwon
    lezwon committed Jan 20, 2021
    Configuration menu
    Copy the full SHA
    432c3c3 View commit details
    Browse the repository at this point in the history

Commits on Jan 29, 2021

  1. formatting

    @Borda @lezwon
    Borda authored and lezwon committed Jan 29, 2021
    Configuration menu
    Copy the full SHA
    beb1ab4 View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'origin/release/1.2-dev' into bugfix/270… 

    …5_weights_tying
    @lezwon
    lezwon committed Jan 29, 2021
    Configuration menu
    Copy the full SHA
    a23e23c View commit details
    Browse the repository at this point in the history

Commits on Feb 1, 2021

  1. update docs

    @lezwon
    lezwon committed Feb 1, 2021
    Configuration menu
    Copy the full SHA
    4cf6799 View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'origin/release/1.2-dev' into bugfix/270… 

    …5_weights_tying
    @lezwon
    lezwon committed Feb 1, 2021
    Configuration menu
    Copy the full SHA
    983619e View commit details
    Browse the repository at this point in the history

Commits on Feb 2, 2021

  1. Merge remote-tracking branch 'origin/release/1.2-dev' into bugfix/270… 

    …5_weights_tying
    @lezwon
    lezwon committed Feb 2, 2021
    Configuration menu
    Copy the full SHA
    7f7e195 View commit details
    Browse the repository at this point in the history

Commits on Feb 17, 2021

  1. Merge branch 'master' of https://github.com/PyTorchLightning/pytorch-… 

    …lightning into bugfix/2705_weights_tying
    @tchaton
    tchaton committed Feb 17, 2021
    Configuration menu
    Copy the full SHA
    7d8823f View commit details
    Browse the repository at this point in the history

  2. update import

    @tchaton
    tchaton committed Feb 17, 2021
    Configuration menu
    Copy the full SHA
    2268df8 View commit details
    Browse the repository at this point in the history

  3. update

    @tchaton
    tchaton committed Feb 17, 2021
    Configuration menu
    Copy the full SHA
    8acfd39 View commit details
    Browse the repository at this point in the history

  4. resolve tests

    Your Name committed Feb 17, 2021
    Configuration menu
    Copy the full SHA
    62042c6 View commit details
    Browse the repository at this point in the history

  5. remove legacy accelerator

    @tchaton
    tchaton committed Feb 17, 2021
    Configuration menu
    Copy the full SHA
    3605c0b View commit details
    Browse the repository at this point in the history