454f620fce3e6fe2f87a91bffd74667d1f8a94f9,slm_lab/agent/net/mlp.py,HydraMLPNet,__init__,#HydraMLPNet#Any#Any#Any#Any#,192

Before Change

                module.cuda()
        self.loss_fn = net_util.get_loss_fn(self, self.loss_spec)
        self.optim = net_util.get_optim(self, self.optim_spec)
        logger.info(f"loss fn: {self.loss_fn}")
        logger.info(f"optimizer: {self.optim}")

    def build_model_heads(self, in_dim):
        """Build each model_head. These are stored as Sequential models in model_heads"""
        assert len(self.head_hid_layers) == len(in_dim), "Hydra head hid_params inconsistent with number in dims"

After Change

                module.cuda()
        self.loss_fn = net_util.get_loss_fn(self, self.loss_spec)
        self.optim = net_util.get_optim(self, self.optim_spec)
        self.lr_decay = getattr(net_util, self.lr_decay)

    def __str__(self):
        return super(HydraMLPNet, self).__str__() + f"\noptim: {self.optim}"

Instances