ff5da2b2bfd03234519927b966923e8baecf6616,slm_lab/agent/net/convnet.py,ConvNet,__init__,#ConvNet#Any#Any#Any#Any#Any#Any#Any#Any#Any#,18

Before Change

                loss_param={"name": "mse_loss"})
        """
        // Calling super on greatgrandfather class
        Module.__init__(self)
        self.in_dim = in_dim
        self.out_dim = out_dim
        self.batch_norm = batch_norm
        self.conv_layers = []

After Change

        self.num_hid_layers = len(self.conv_layers) + len(self.flat_layers) - 1
        self.init_params()
        // Init other net variables
        self.params = list(self.conv_model.parameters()) + \
            list(self.dense_model.parameters())
        self.optim = net_util.get_optim_multinet(self.params, optim_param)
        self.loss_fn = net_util.get_loss_fn(self, loss_param)
        self.clamp_grad = clamp_grad
        self.clamp_grad_val = clamp_grad_val

Instances