pytorch-lightning
28dbfeeb - Set `_DeviceDtypeModuleMixin` _device from torch's default device function (#21164)

Commit
102 days ago
Set `_DeviceDtypeModuleMixin` _device from torch's default device function (#21164) * fix: set `_DeviceDtypeModuleMixin` _device from torch's default device function. * add torch 2.2.2 or below compatibility * add import for `_TORCH_GREATER_EQUAL_2_3 * fix: restore torch default dtype once test_submodules_context_device_and_dtype is finished. --------- Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> Co-authored-by: Bhimraj Yadav <bhimrajyadav977@gmail.com> Co-authored-by: Nicki Skafte Detlefsen <skaftenicki@gmail.com>
Author
Parents
Loading