Issue-Report: PyTroch v. 0.2.1.dev240521 - Dropout no longer working

microsoft / DirectML

DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.

MIT License

2.21k stars 294 forks source link

Issue-Report: PyTroch v. 0.2.1.dev240521 - Dropout no longer working #590

Closed mor-lu closed 4 months ago

mor-lu commented 4 months ago

Since release 0.2.1.dev240521 when using a dropout layer the following error displays and the whole model is trained on the cpu:

'aten:native.dropout' is not currently supported on the DML backend and will fall back to run on the CPU.

Reverting to the previous version 0.2.0.dev230426 fixed the issue.

Windows 11 AMD Radeon rx7600 GPU

joshjkim commented 4 months ago

Hi @mor-lu , thank you for the feedback. We've now added support for native_dropout in our latest torch-directml 0.2.2.dev240614 release. Please run pip install torch-directml --upgrade to update to this version