Align autoencoder splits and add multi-GPU training

This commit is contained in:
2026-07-09 13:31:24 +08:00
parent 8a083ed538
commit 54fedc620b
2 changed files with 83 additions and 30 deletions

View File

@@ -64,9 +64,12 @@ Pretrain the exposure encoder as a denoising autoencoder using training-set EIDs
```bash
python train_exposure_autoencoder.py \
--exposure_cache_dir ukb_exposure_cache \
--train_eid_file ukb_train_eid.csv
--train_eid_file ukb_train_eid.csv \
--val_eid_file ukb_val_eid.csv
```
The best checkpoint contains both `model_state_dict`, an `encoder_state_dict`
compatible with the default gated `TimesNetExposureEncoder`, and the channel
normalization statistics needed when the encoder is attached to DeepHealth.
Multi-GPU pretraining follows the main trainer interface: add
`--data_parallel --gpu_ids 0,1,2,3`.