-
- Downloads
Merge branch 'add-datamodule-andre' into 'add-datamodule'
Reviewed DataModule design+docs+types See merge request biosignal/software/ptbench!7
No related branches found
No related tags found
This commit is part of merge request !6. Comments created here will be created in the context of that merge request.
Showing
- src/ptbench/engine/callbacks.py 356 additions, 108 deletionssrc/ptbench/engine/callbacks.py
- src/ptbench/engine/device.py 150 additions, 0 deletionssrc/ptbench/engine/device.py
- src/ptbench/engine/trainer.py 61 additions, 117 deletionssrc/ptbench/engine/trainer.py
- src/ptbench/models/alexnet.py 170 additions, 50 deletionssrc/ptbench/models/alexnet.py
- src/ptbench/models/densenet.py 172 additions, 51 deletionssrc/ptbench/models/densenet.py
- src/ptbench/models/loss_weights.py 70 additions, 0 deletionssrc/ptbench/models/loss_weights.py
- src/ptbench/models/normalizer.py 59 additions, 22 deletionssrc/ptbench/models/normalizer.py
- src/ptbench/models/pasa.py 183 additions, 103 deletionssrc/ptbench/models/pasa.py
- src/ptbench/scripts/train.py 63 additions, 70 deletionssrc/ptbench/scripts/train.py
- src/ptbench/scripts/utils.py 57 additions, 0 deletionssrc/ptbench/scripts/utils.py
- src/ptbench/utils/accelerator.py 0 additions, 127 deletionssrc/ptbench/utils/accelerator.py
- src/ptbench/utils/image.py 36 additions, 0 deletionssrc/ptbench/utils/image.py
- src/ptbench/utils/resources.py 159 additions, 147 deletionssrc/ptbench/utils/resources.py
Loading
Please register or sign in to comment