Skip to content

Can't convert cuda:0 device type tensor to numpy #7

@GZWSAMA

Description

@GZWSAMA

Modify the code here to original_sizes = original_sizes.cpu().numpy() sam

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