Skip to content

map_location with device CPU should be provided in case --no-gpu option is used by the executer  #35

@harshlunia7

Description

@harshlunia7

load_dict(generator, pretrained_net)

torch.load(opt.load_name, map_location=torch.device('cpu'))

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