ml-workflow
stable
Contents:
workflow package
workflow.functional package
workflow.torch package
workflow.ignite package
ml-workflow
»
Module code
»
workflow.torch.module_device
Source code for workflow.torch.module_device
[docs]
def
module_device
(
module
):
return
next
(
module
.
parameters
())
.
device