Go to file

Jiarui Li d760c45baf feat: Add multi-GPU training and improve config/ignore

Add train_multigpu.py for distributed data parallel training.

Update train.py to save the training configuration to a JSON file.

Generalize .gitignore to exclude all *.pt checkpoint files.

Delete obsolete train_dpp.py file.

2025-10-17 14:09:34 +08:00

.gitattributes

feat: Add training and validation data via Git LFS

2025-10-16 14:24:56 +08:00

.gitignore

feat: Add multi-GPU training and improve config/ignore

2025-10-17 14:09:34 +08:00

GEMINI.md

feat: Implement time-aware GPT-2 for patient event prediction

2025-10-16 14:21:36 +08:00

LICENSE

Initial commit

2025-10-15 13:54:52 +08:00

models.py

feat: Add covariate-aware model and piecewise encoder

2025-10-17 12:04:50 +08:00

prepare_data.py

feat: Implement time-aware GPT-2 for patient event prediction

2025-10-16 14:21:36 +08:00

README.md

Initial commit

2025-10-15 13:54:52 +08:00

requirements.txt

refactor: Use AdamW optimizer and increase early stopping patience

2025-10-17 10:31:12 +08:00

train_multigpu.py

feat: Add multi-GPU training and improve config/ignore

2025-10-17 14:09:34 +08:00

train.py

feat: Add multi-GPU training and improve config/ignore

2025-10-17 14:09:34 +08:00

ukb_real_train.bin

feat: Add training and validation data via Git LFS

2025-10-16 14:24:56 +08:00

ukb_real_val.bin

feat: Add training and validation data via Git LFS

2025-10-16 14:24:56 +08:00

utils.py

feat: Implement time-aware GPT-2 for patient event prediction

2025-10-16 14:21:36 +08:00

README.md

DeepHealth