Skip to content
Snippets Groups Projects
Commit 11b65a40 authored by Daniel CARRON's avatar Daniel CARRON :b: Committed by André Anjos
Browse files

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 9eb1e023
No related branches found
No related tags found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment