-
- Downloads
Updates to the data module and models
parent
9e721cb3
Branches
Tags
This commit is part of merge request !7. Comments created here will be created in the context of that merge request.
Showing
- src/ptbench/data/datamodule.py 71 additions, 62 deletionssrc/ptbench/data/datamodule.py
- src/ptbench/engine/trainer.py 28 additions, 19 deletionssrc/ptbench/engine/trainer.py
- src/ptbench/models/densenet.py 15 additions, 0 deletionssrc/ptbench/models/densenet.py
- src/ptbench/models/pasa.py 10 additions, 0 deletionssrc/ptbench/models/pasa.py
- src/ptbench/scripts/train.py 22 additions, 39 deletionssrc/ptbench/scripts/train.py
Loading
Please register or sign in to comment