901cc588fee90215d2e2831bd4a3e7481b307ba7,slm_lab/agent/algorithm/dqn.py,HydraDQN,init_nets,#HydraDQN#,408

Before Change

        self.target_net = NetClass(self.net_spec, self, in_dims, out_dims)
        self.online_net = self.target_net
        self.eval_net = self.target_net
        logger.info(f"Training on gpu: {self.net.gpu}")

    @lab_api
    def calc_pdparam(self, x, evaluate=True):
        """

After Change

        NetClass = getattr(net, self.net_spec["type"])
        self.net = NetClass(self.net_spec, self, in_dims, out_dims)
        self.target_net = NetClass(self.net_spec, self, in_dims, out_dims)
        self.net_names = ["net", "target_net"]
        self.post_init_nets()
        self.online_net = self.target_net
        self.eval_net = self.target_net

    @lab_api

Instances