Method eval

cdslib/core/models/base_model.py:159–167

Set all the nn.modules to evaluation mode.

(self)

            v.train()

    def eval(self):
        """Set all the nn.modules to evaluation mode."""
        vs = inspect.getmembers(self, lambda v: isinstance(v, BaseModel))
        for v_name, v in vs:
            v.eval()

        vs = inspect.getmembers(self, lambda v: isinstance(v, torch.nn.Module))
        for v_name, v in vs:
            v.eval()

    def load_state_dict(
        self,