Method add_summary

axlearn/common/module.py:1114–1121

Adds the named value to `OutputCollection.summaries`. Args: name: The name of the item to add. value: The value to add.

(self, name: str, value: Union[Summable, Tensor, Summary])

    def is_training(self) -> bool:
        return self.get_invocation_context().is_training

    @property
    def prng_key(self) -> Tensor:
        return self.get_invocation_context().prng_key

    @property
    def state(self):
        return self.get_invocation_context().state

    def add_summary(self, name: str, value: Union[Summable, Tensor, Summary]):
        """Adds the named value to `OutputCollection.summaries`.