Skip to content

Make Pose Classifier Device Agnostic #2

@Log45

Description

@Log45

In nn_pose_classifier.py: remove device as an object attribute and instead pass device as a parameter in each inference or training method to avoid saving the model with a device attached to it (causing issues when loading it on a different device).

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