MCPcopy Create free account
hub / github.com/apple/axlearn / add_module_output

Method add_module_output

axlearn/common/module.py:1127–1128  ·  view source on GitHub ↗
(self, name: str, value: Tensor)

Source from the content-addressed store, hash-verified

1125
1126 Args:
1127 name: The name of the item to add.
1128 value: The value to add.
1129 """
1130 return self.get_invocation_context().add_summary(name, value)
1131

Callers 15

forwardMethod · 0.45
forwardMethod · 0.45
forwardMethod · 0.45
_metricsMethod · 0.45
forwardMethod · 0.45
extend_stepMethod · 0.45
forwardMethod · 0.45
predictMethod · 0.45
forwardMethod · 0.45
forwardMethod · 0.45
forwardMethod · 0.45
_forward_for_modeMethod · 0.45

Calls 1

Tested by 9

forwardMethod · 0.36
_metricsMethod · 0.36
forwardMethod · 0.36
forwardMethod · 0.36
predictMethod · 0.36
forwardMethod · 0.36
forwardMethod · 0.36
incMethod · 0.36
fnMethod · 0.36