Closed mor-lu closed 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
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