src/ptbench/configs/models/pasa.py · bf987617bd745a7d8d92a50c921a675638ed7f9c · medai / software / mednet

1 year ago

Functional model training step for pasa · bf987617

Daniel CARRON authored 1 year ago

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

bf987617

History

Functional model training step for pasa

Daniel CARRON authored 1 year ago

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

pasa.py 1.11 KiB