Method set

axlearn/common/layers.py:110–116

(self, **kwargs)

        def set(self, **kwargs) -> "RedirectToSharedModule.Config":
            try:
                super().set(**kwargs)
            except UnknownFieldError as e:
                logging.info("Ignoring %s", e)
                # We intentionally ignore this exception.
            return self

    def _methods_to_wrap_for_auto_child_context(self) -> dict[str, Callable]:
        cfg: RedirectToSharedModule.Config = self.config
        method_dict = {}
        for source_method, target_method in (cfg.method_map or {"forward": "forward"}).items():