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)