Skip to content

Error in human and garment pose training scripts #52

@AnasKunda

Description

@AnasKunda

There is a small bug in train.py script of human and garment pose models.

The bug is present in below lines:

for test_keypoints in test_dataloader:
            model.eval()
            test_predictions, _ = model(test_keypoints)

The correction should be:

for test_keypoints, _ in test_dataloader:
            model.eval()
            test_predictions, _ = model(test_keypoints)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions