GitLab

If there are less than half of model_cfg.NUM_KEYPOINTS it will lead to 
RuntimeError: unsupported operation: some elements of the input tensor and the written-to tensor refer to a single memory location. 
Please clone() the tensor before performing the operation.