怎么避免UserWarning The operator atenbinary_cross_entropy is not currently supported on the DML backend and will fall back to run on the CPU This may have performance implications Triggered internally at
这个警告是因为使用的运算符 'aten::binary_cross_entropy' 目前不支持 DML 后端,因此会回退到在 CPU 上运行。这可能会影响性能。要避免这个警告,可以尝试在代码中将 DML 后端更改为其他后端,例如 CUDA 或 CPU 后端。另外,也可以尝试更新 PyTorch 版本,以查看是否已更新支持 DML 后端的运算符。
原文地址: https://www.cveoy.top/t/topic/dvHd 著作权归作者所有。请勿转载和采集!