Summary: 5 instances, 5 unique

Text	Count
# TODO: fix this type when PyTorch fixes theirs	1
#TODO: try delay_allreduce=True	1
prediction_reorien_adv.requires_grad_() #TODO Might not be required	1
shift = [(dim + 1) // 2 for dim in x.shape] #TODO: looks wrong	1
TODO: I need to make this more generic and reusable	1