【pytorch】图像运行过程中,保证梯度情况下变换

发布时间:2023年12月18日

部分操作是危险的,会中断梯度流。

self.patch_transformer(adv_patch, lab_batch, img_size, do_rotate=True, rand_loc=False)
p_img_batch = self.patch_applier(img_batch, adv_batch_t)    # torch.Size([56, 3, 329, 416])

可行危险操作

torch.clamp(adv_batch, 0.000001, 0.99999)
affine

文章来源:https://blog.csdn.net/prinTao/article/details/135049839
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。