Merge branch '27-add-support-for-torch-2-0' into 'develop' (861d559b) · Commits · Magnet / DecLearn / declearn2 · GitLab

Mentions légales du service

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

D

declearn2
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Terraform modules
- Service Desk

Snippets Groups Projects

Admin message

A GitLab upgrade is scheduled for Monday, April 7, 2025. Service interruptions are expected in the morning. We'll keep you posted on the progress of the upgrade on the Mattermost channel: https://mattermost.inria.fr/devel/channels/gitlab. We recommend that you do not work on the platform until an announcement indicates that maintenance is complete.

861d559b

Commit 861d559b authored 1 year ago by ANDREY Paul

Downloads
- Plain Diff

Merge branch '27-add-support-for-torch-2-0' into 'develop'

Add support for Torch 2.0

Closes #27

See merge request !49

parents 9f0035d7 6d3695fd

No related branches found

No related tags found

Loading

Checking pipeline status

Changes 10
Pipelines 1

Hide whitespace changes

Inline Side-by-side

Showing

declearn/communication/__init__.py 2 additions, 0 deletions

declearn/communication/__init__.py
declearn/model/torch/_model.py 89 additions, 90 deletions

declearn/model/torch/_model.py
declearn/model/torch/_samplewise/__init__.py 78 additions, 0 deletions

declearn/model/torch/_samplewise/__init__.py
declearn/model/torch/_samplewise/functorch.py 96 additions, 0 deletions

declearn/model/torch/_samplewise/functorch.py
declearn/model/torch/_samplewise/shared.py 56 additions, 0 deletions

declearn/model/torch/_samplewise/shared.py
declearn/model/torch/_samplewise/torchfunc.py 80 additions, 0 deletions

declearn/model/torch/_samplewise/torchfunc.py
docs/setup.md 7 additions, 0 deletions

docs/setup.md
pyproject.toml 15 additions, 5 deletions

pyproject.toml
scripts/run_tests.sh 28 additions, 0 deletions

scripts/run_tests.sh
test/model/test_torch.py 17 additions, 4 deletions

test/model/test_torch.py

with 468 additions and 99 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment