banding_removal/fastmri/orientation_adversary/adversary_mixin.py (2 lines): - line 64: TODO: I need to make this more generic and reusable - line 151: prediction_reorien_adv.requires_grad_() #TODO Might not be required fastmri/models/varnet.py (1 line): - line 91: # TODO: fix this type when PyTorch fixes theirs banding_removal/fastmri/distributed_mixin.py (1 line): - line 117: #TODO: try delay_allreduce=True banding_removal/fastmri/data/transforms.py (1 line): - line 534: shift = [(dim + 1) // 2 for dim in x.shape] #TODO: looks wrong