hub / github.com/deepspeedai/DeepSpeed / debug_param2name_id_shape_device

Function debug_param2name_id_shape_device

deepspeed/utils/debug.py:76–77 · view source on GitHub ↗

(param)

Source from the content-addressed store, hash-verified

74
75
76	def debug_param2name_id_shape_device(param):
77	return f"name={debug_param2name(param)} id={ds_id(param)} shape={ds_shape(param)} device={param.device}"
78
79
80	def debug_param2name_id_numel(param):

partitionMethod · 0.90

partition_gradientsMethod · 0.90

debug_param2nameFunction · 0.85

ds_idFunction · 0.85

ds_shapeFunction · 0.85

no test coverage detected