Skip to content

[Solution] Unable to continue training model on multi_gpu #21

New issue
New issue
Open
Open
[Solution] Unable to continue training model on multi_gpu#21
@MaxShi007

Description

@MaxShi007
MaxShi007
opened on Jun 9, 2022

model.module.load_state_dict(torch.load(args.pretrain))

PointPWC/train.py

Line 116 in f882e4c

model.load_state_dict(torch.load(args.pretrain))

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