[ptbench.engine.device] Move AcceleratorProcessor to engine submodule, revamp...
[ptbench.engine.device] Move AcceleratorProcessor to engine submodule, revamp it, and reflect changes in trainer
parent
a518c23d
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/device.py 150 additions, 0 deletionssrc/ptbench/engine/device.py
- src/ptbench/engine/trainer.py 23 additions, 16 deletionssrc/ptbench/engine/trainer.py
- src/ptbench/scripts/train.py 6 additions, 6 deletionssrc/ptbench/scripts/train.py
- src/ptbench/utils/accelerator.py 0 additions, 131 deletionssrc/ptbench/utils/accelerator.py
Loading
Please register or sign in to comment