def __init__(self, action_dim, body, hidden, dist, particles):
    super(DuelingHyperNet, self).__init__()
    self.mixer = False
    self.config = DuelingNet_config(body.feature_dim, action_dim)
    self.config['fc_value'] = self.config['fc_value']._replace(d_hidden=hidden)
    self.config['fc_advantage'] = self.config['fc_advantage']._replace(d_hidden=hidden)
    self.fc_value = LinearGenerator(self.config['fc_value']).cuda()
    self.fc_advantage = LinearGenerator(self.config['fc_advantage']).cuda()
    self.features = body
    self.s_dim = self.config['s_dim']
    self.z_dim = self.config['z_dim']
    self.n_gen = self.config['n_gen']
    self.particles = particles
    self.noise_sampler = NoiseSampler(dist, self.z_dim, self.particles)
    self.to(Config.DEVICE)
def __init__(self, action_dim, body, hidden, dist, particles):
    super(DuelingHyperNet, self).__init__()
    self.mixer = False
    self.config = DuelingNet_config(body.feature_dim, action_dim)
    self.config['fc_value'] = self.config['fc_value']._replace(d_hidden=hidden)
    self.config['fc_advantage'] = self.config['fc_advantage']._replace(d_hidden=hidden)
    self.fc_value = LinearGenerator(self.config['fc_value']).cuda()
    self.fc_advantage = LinearGenerator(self.config['fc_advantage']).cuda()
    self.features = body
    self.s_dim = self.config['s_dim']
    self.z_dim = self.config['z_dim']
    self.n_gen = self.config['n_gen']
    self.particles = particles
    self.noise_sampler = NoiseSampler(dist, self.z_dim, self.particles)
    self.to(Config.DEVICE)