【已解决】RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Float‘
问题描述最近在复现Watermark-Robustness-Toolbox的时候出现了一些问题,并通过不懈的努力(不是)把这个给完美的跑起来了,这很“完美”。先来简单介绍一下,这是一个攻防项目,github链接为:https://github.com/dnn-security/Watermark-Robustness-Toolbox,论文链接为:https://arxiv.org/abs/2108