Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

mednet
- Activity
- Members
- Labels
- Environments
- Terraform modules
- Incidents

Snippets Groups Projects

bd79cd2f

Commit bd79cd2f authored 1 year ago by Daniel CARRON

Committed by André Anjos 1 year ago

Downloads
- Patches
- Plain Diff

Removed reliance on make_dataset and added method to cache samples for default shenzhen

parent 9a03daba

Branches

Tags

1 merge request!6Making use of LightningDataModule and simplification of data loading

Changes 12

Hide whitespace changes

Inline Side-by-side

Showing

pyproject.toml 1 addition, 1 deletion

pyproject.toml
src/ptbench/configs/datasets/__init__.py 0 additions, 1 deletion

src/ptbench/configs/datasets/__init__.py
src/ptbench/configs/datasets/shenzhen/__init__.py 3 additions, 0 deletions

src/ptbench/configs/datasets/shenzhen/__init__.py
src/ptbench/configs/datasets/shenzhen/default.py 24 additions, 5 deletions

src/ptbench/configs/datasets/shenzhen/default.py
src/ptbench/configs/models/pasa.py 7 additions, 1 deletion

src/ptbench/configs/models/pasa.py
src/ptbench/data/__init__.py 34 additions, 28 deletions

src/ptbench/data/__init__.py
src/ptbench/data/dataset.py 4 additions, 1 deletion

src/ptbench/data/dataset.py
src/ptbench/data/loader.py 9 additions, 1 deletion

src/ptbench/data/loader.py
src/ptbench/data/shenzhen/__init__.py 22 additions, 36 deletions

src/ptbench/data/shenzhen/__init__.py
src/ptbench/data/transforms.py 4 additions, 1 deletion

src/ptbench/data/transforms.py
src/ptbench/models/pasa.py 13 additions, 2 deletions

src/ptbench/models/pasa.py
src/ptbench/scripts/train.py 10 additions, 0 deletions

src/ptbench/scripts/train.py

with 131 additions and 77 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