Removed commented code of 3d torch fix