742c48e994
* Ensure we move the model to eval mode before running evaluation * Ensure we set the flag appropriately across all stages * Add test, move hooks logic * Apply same fix to the validate loop * Update pytorch_lightning/trainer/trainer.py * Fix function name * Fix order, add predict * Shorten the name * Fix input dm, drop duplicate on predict start hook call, as it's called in the setup function * Use hook, remove double call |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| datamodule.py | ||
| decorators.py | ||
| grads.py | ||
| hooks.py | ||
| lightning.py | ||
| memory.py | ||
| optimizer.py | ||
| saving.py | ||
| step_result.py | ||