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.
17 lines
186 B
Plaintext
17 lines
186 B
Plaintext
# IDE settings
|
|
.idea/
|
|
|
|
# Python cache
|
|
__pycache__/
|
|
|
|
# Model checkpoints
|
|
*.pt
|
|
|
|
# Large data files
|
|
ukb_delphi.txt
|
|
ukb_real.bin
|
|
|
|
# Small data files
|
|
fields.txt
|
|
icd10_codes_mod.tsv
|
|
labels.csv |