Merge branch 'loss-balancing' into 'main'
Replace sampler balancing by loss balancing Closes #6 See merge request biosignal/software/mednet!38
No related branches found
No related tags found
Showing
- doc/api.rst 1 addition, 0 deletionsdoc/api.rst
- src/mednet/config/models/alexnet.py 1 addition, 2 deletionssrc/mednet/config/models/alexnet.py
- src/mednet/config/models/alexnet_pretrained.py 1 addition, 2 deletionssrc/mednet/config/models/alexnet_pretrained.py
- src/mednet/config/models/densenet.py 1 addition, 2 deletionssrc/mednet/config/models/densenet.py
- src/mednet/config/models/densenet_pretrained.py 1 addition, 2 deletionssrc/mednet/config/models/densenet_pretrained.py
- src/mednet/config/models/densenet_rs.py 1 addition, 2 deletionssrc/mednet/config/models/densenet_rs.py
- src/mednet/config/models/pasa.py 1 addition, 2 deletionssrc/mednet/config/models/pasa.py
- src/mednet/data/datamodule.py 4 additions, 44 deletionssrc/mednet/data/datamodule.py
- src/mednet/engine/callbacks.py 1 addition, 0 deletionssrc/mednet/engine/callbacks.py
- src/mednet/engine/trainer.py 2 additions, 0 deletionssrc/mednet/engine/trainer.py
- src/mednet/models/alexnet.py 16 additions, 65 deletionssrc/mednet/models/alexnet.py
- src/mednet/models/densenet.py 16 additions, 64 deletionssrc/mednet/models/densenet.py
- src/mednet/models/loss_weights.py 142 additions, 49 deletionssrc/mednet/models/loss_weights.py
- src/mednet/models/model.py 173 additions, 0 deletionssrc/mednet/models/model.py
- src/mednet/models/pasa.py 16 additions, 82 deletionssrc/mednet/models/pasa.py
- src/mednet/scripts/train.py 2 additions, 4 deletionssrc/mednet/scripts/train.py
- tests/test_cli.py 2 additions, 4 deletionstests/test_cli.py
Loading
Please register or sign in to comment