Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

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

Snippets Groups Projects

bf987617

Commit bf987617 authored 1 year ago by Daniel CARRON

Downloads
- Patches
- Plain Diff

Functional model training step for pasa

Implemented BCEWithLogitsLoss reweighting function
Removed save_hyperparameters from model
Apply augmentation transforms on singular images
Fixed model summary

parent bd46b6bd

Branches

Tags

Loading

Checking pipeline status

Changes 6
Pipelines 1

Hide whitespace changes

Inline Side-by-side

Showing

src/ptbench/configs/models/pasa.py 8 additions, 5 deletions

src/ptbench/configs/models/pasa.py
src/ptbench/data/datamodule.py 10 additions, 2 deletions

src/ptbench/data/datamodule.py
src/ptbench/data/dataset.py 6 additions, 43 deletions

src/ptbench/data/dataset.py
src/ptbench/engine/callbacks.py 1 addition, 3 deletions

src/ptbench/engine/callbacks.py
src/ptbench/engine/trainer.py 8 additions, 4 deletions

src/ptbench/engine/trainer.py
src/ptbench/models/pasa.py 45 additions, 16 deletions

src/ptbench/models/pasa.py

with 78 additions and 73 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