Fix dump of search job checkpoints
@@ -90,6 +90,7 @@ def _dump_checkpoint(args):
# Dump it
print(f"# Dump of checkpoint: {checkpoint_file}")
+ if "model" in checkpoint:
print(f"parameter_names: {list(checkpoint['model'][0].keys())}")
excluded_keys = {"model", "optimizer_state_dict"}
if args.keys is not None:
