Solved by dhecloud in post #2 
              
              
                The issue was actually a really silly one. When i loaded the ground truths in the dataset, it was loaded into a tensor. For example, 
self.all_joints = read_joints()
other reading stuff

When i loaded the data in the __getitem__ function, i would perform random translation/scaling/rotating and norma…