Train myself on data

[Sweetest316]

Hi!
First thanks for your work that show magnificient results.
Hello, I want to train my own data set with your model (only 4 key points), I have converted the data to coco key points format, num_joints changed to 4, but when training to read the data:
offset, offset_weight = self.offset_generator(joints_list[0], area) returns the offset and offset_weight values to 0, and the inference result is also wrong, may I ask where I set wrong?