Add resumable exposure autoencoder training

This commit is contained in:
2026-07-10 05:32:24 +08:00
parent e439c3c98c
commit 75da450720
2 changed files with 240 additions and 34 deletions

View File

@@ -83,6 +83,16 @@ torchrun --standalone --nproc_per_node=4 train_exposure_autoencoder.py \
In DDP mode, `--batch_size` is the global batch size and must be divisible by
the number of processes.
The trainer also writes `last.pt` after every epoch so interrupted runs can be
continued. Pass the run directory to reuse the original `train_config.json`;
the trainer will load `last.pt` when available and fall back to `best.pt` for
older runs:
```bash
python train_exposure_autoencoder.py \
--resume_checkpoint runs/exposure_ae_RUN
```
Encode every cached exposure window once:
```bash