Source code for workflow.torch.module_device




[docs]def module_device(module):
    return next(module.parameters()).device