[engine.predictor] Allow prediction to work when batch-size!=0; Discard CSVs...
[engine.predictor] Allow prediction to work when batch-size!=0; Discard CSVs and use JSON for output; Implement lightning logging like in training; Separates batched (collated) data at prediction; Adapts all models to new paradigm; Remove CSVPredictionsWriter; Adapt predict script to all changes
parent
b0655852
No related branches found
No related tags found
Showing
- src/ptbench/engine/callbacks.py 0 additions, 49 deletionssrc/ptbench/engine/callbacks.py
- src/ptbench/engine/predictor.py 37 additions, 31 deletionssrc/ptbench/engine/predictor.py
- src/ptbench/engine/trainer.py 1 addition, 1 deletionsrc/ptbench/engine/trainer.py
- src/ptbench/models/alexnet.py 3 additions, 11 deletionssrc/ptbench/models/alexnet.py
- src/ptbench/models/densenet.py 3 additions, 11 deletionssrc/ptbench/models/densenet.py
- src/ptbench/models/logistic_regression.py 5 additions, 12 deletionssrc/ptbench/models/logistic_regression.py
- src/ptbench/models/mlp.py 5 additions, 12 deletionssrc/ptbench/models/mlp.py
- src/ptbench/models/pasa.py 3 additions, 11 deletionssrc/ptbench/models/pasa.py
- src/ptbench/models/separate.py 29 additions, 0 deletionssrc/ptbench/models/separate.py
- src/ptbench/scripts/predict.py 74 additions, 38 deletionssrc/ptbench/scripts/predict.py
Loading
Please register or sign in to comment