Andre Anjos
authored
[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
| Name | Last commit | Last update |
|---|---|---|
| .. | ||
| loggers | ||
| __init__.py | ||
| callbacks.py | ||
| device.py | ||
| evaluator.py | ||
| predictor.py | ||
| trainer.py | ||