Merge branch 'xju-upgrade-pytorch' into 'dev'
Upgrade to Pytorch 2.0 See merge request gnn4itkteam/acorn!81
Showing
- .gitignore 1 addition, 0 deletions.gitignore
- .gitlab/pipelines/.test-gitlab-ci.yaml 6 additions, 3 deletions.gitlab/pipelines/.test-gitlab-ci.yaml
- Dockerfile 6 additions, 6 deletionsDockerfile
- README.md 43 additions, 9 deletionsREADME.md
- acorn/core/core_utils.py 2 additions, 3 deletionsacorn/core/core_utils.py
- acorn/stages/data_reading/data_reading_stage.py 1 addition, 3 deletionsacorn/stages/data_reading/data_reading_stage.py
- acorn/stages/edge_classifier/edge_classifier_stage.py 10 additions, 5 deletionsacorn/stages/edge_classifier/edge_classifier_stage.py
- acorn/stages/edge_classifier/models/gnn_submodule/updater.py 6 additions, 1 deletionacorn/stages/edge_classifier/models/gnn_submodule/updater.py
- acorn/stages/edge_classifier/models/utils.py 9 additions, 6 deletionsacorn/stages/edge_classifier/models/utils.py
- acorn/stages/graph_construction/models/metric_learning.py 12 additions, 5 deletionsacorn/stages/graph_construction/models/metric_learning.py
- acorn/stages/graph_construction/utils.py 18 additions, 15 deletionsacorn/stages/graph_construction/utils.py
- acorn/stages/track_building/utils.py 17 additions, 7 deletionsacorn/stages/track_building/utils.py
- acorn/utils/eval_utils.py 3 additions, 2 deletionsacorn/utils/eval_utils.py
- acorn/utils/loading_utils.py 29 additions, 22 deletionsacorn/utils/loading_utils.py
- acorn/utils/mapping_utils.py 2 additions, 1 deletionacorn/utils/mapping_utils.py
- acorn/utils/version_utils.py 22 additions, 0 deletionsacorn/utils/version_utils.py
- check_acorn.py 89 additions, 0 deletionscheck_acorn.py
- examples/CTD_2023/gnn_eval.yaml 2 additions, 1 deletionexamples/CTD_2023/gnn_eval.yaml
- examples/CTD_2023/gnn_infer.yaml 2 additions, 1 deletionexamples/CTD_2023/gnn_infer.yaml
- examples/CTD_2023/gnn_train.yaml 2 additions, 1 deletionexamples/CTD_2023/gnn_train.yaml
Loading
Please register or sign in to comment