File size: 156 Bytes
b9a3411 | 1 2 3 4 5 6 7 | def get_parameter_device(module):
return next(module.parameters()).device
def get_parameter_dtype(module):
return next(module.parameters()).dtype
|
b9a3411 | 1 2 3 4 5 6 7 | def get_parameter_device(module):
return next(module.parameters()).device
def get_parameter_dtype(module):
return next(module.parameters()).dtype
|