Skip to content

Potential issues if multiple checkpoints for one config #19

Closed
@eddiebergman

Description

@eddiebergman

ckpt_files = glob.glob(str(checkpoint_dir / "*.ckpt"))
if ckpt_files:
# Load the checkpoint and retrieve necessary data
checkpoint_path = ckpt_files[-1]
checkpoint = torch.load(checkpoint_path)

This just grabs the last checkpoint found by glob which gives no guaranteed order of what it returns. For now it might be safer to just error if multiple are found. Not sure about a long term solution

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    Status

    Done

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions