Skip to content

how to modify multigpu_train.py to train on cpu with custom dataset #396

Open
@vidkr

Description

@vidkr

how to modify multigpu_train.py to train on cpu with custom dataset. I create a dataset of my own , but getting error while trying to run multigpu_train.py. i attached a file with this where you can see error and warning gererated while running multi_gpu training. Help me to solve these errors
model_issue_output.pdf

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