Add resumable exposure autoencoder training
This commit is contained in:
10
README.md
10
README.md
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user