Skip to content

Misplaced parenthesis #456

Closed
Closed
@BuissonFlorent

Description

@BuissonFlorent

https://github.com/rasbt/LLMs-from-scratch/blob/bb31de89993441224e9005926dedad95395bb058/ch05/01_main-chapter-code/gpt_train.py#L242C4-L242C70

I believe the last line in the script should read
model.load_state_dict(torch.load("model.pth", weights_only=True))
With weights_only as a parameter of torch.load(), not model.load_state_dict()

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions