Heimdall.trainer.HeimdallTrainer.CHECKPOINT_KEYS

HeimdallTrainer.CHECKPOINT_KEYS = ('tasks', 'fg', 'fe', 'fc', 'model', 'dataset.preprocess_args.data_path')