JiaruiLi/DeepHealth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Clarify empty extra-info logging

Browse Source
This commit is contained in:
Jiarui Li
2026-06-29 11:26:48 +08:00
parent d4a29063d6
commit ca294193f4
3 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
train_next_step.py
View File

@@ -30,6 +30,7 @@ from targets import CHECKUP_IDX, NO_EVENT_IDX, PAD_IDX
from train_util import (
configure_torch_for_training,
create_unique_run_dir,
format_extra_info_types,
load_extra_info_types_file,
resolve_device,
save_checkpoint,
@@ -526,7 +527,7 @@ def main() -> None:
logger.info(f"Starting next-step training run: {run_name}")
logger.info(f"Device: {device}")
logger.info(f"extra_info_types: {args.extra_info_types or 'all'}")
logger.info(f"extra_info_types: {format_extra_info_types(args.extra_info_types)}")
logger.info(f"readout={args.readout_name}, target_mode={args.target_mode}")
dataset = HealthDataset(