RuntimeError: cuFFT only supports dimensions whose sizes are powers of two when computing in half precision, but got a signal size of[8, 750]

这个错误表明在使用半精度计算(half precision)时,cuFFT 仅支持尺寸为二的幂的维度。然而,你的输入信号的尺寸为 [8, 750],并不符合这个要求。

解决办法是不启用半精度计算。

在json文件中设置对应参数false。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐