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

Before Change

        self.out_dim = out_dim
        self.batch_norm = batch_norm
        self.conv_layers = []
        self.batch_norms = []
        self.flat_layers = []
        self.build_conv_layers(conv_hid)
        self.build_flat_layers(flat_hid, out_dim)
        self.num_hid_layers = len(self.conv_layers) + len(self.flat_layers)

After Change

        self.conv_model = self.build_conv_layers(
            hid_layers[0], hid_layers_activation)
        self.flat_layers = []
        self.dense_model = self.build_flat_layers(
            hid_layers[1], out_dim, hid_layers_activation)
        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