RobustBench / robustbench

RobustBench: a standardized adversarial robustness benchmark [NeurIPS 2021 Benchmarks and Datasets Track]
https://robustbench.github.io
Other
664 stars 99 forks source link

Fourier transform is not supported? #178

Closed tuningManBin closed 7 months ago

tuningManBin commented 7 months ago

Hello, thanks for providing the robust testing tool. Fourier transform torch.fft.fft is used in my model (Complex numbers are extracted into two channels [tensor.real, tensor.imag]). Attacks can be carried out sequentially in stages apgd-ce, apgd-t, and fab-t. But when conducting a square attack, an error occurs. cuFFT error: CUFFT_INVALID_SIZE. I am confused whether autoattack does not support the Fourier transform.

tuningManBin commented 7 months ago

I checked the code and found the reason: Infinity appears in the predicted value