Method forward

axlearn/common/module_test.py:72–73

(self)

        class _ImageModule(Module):
            def forward(self):
                self.add_summary("img0", ImageSummary(val))

        module = _ImageModule.default_config().set(name="tmp").instantiate(parent=None)