Janmaking

270-280 with torch.set_grad_enabled(True): grid, points_f, points_m, points_a = registration_model( img_f, img_m, lmbda, True ) ## img_a = align_img(grid, img_m) img_a = align_img(grid, img_m[0]) if seg_available: seg_a = align_img(grid, seg_m)