Align autoencoder splits and add multi-GPU training
This commit is contained in:
@@ -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`.
|
||||
|
||||
Reference in New Issue
Block a user