예제 #1
0
 def __init__(self, sess, state_dim, action_dim, action_bound,
              learning_rate, tau, batch_size, predictor_type,
              use_batch_norm):
     self.predictor_type = predictor_type
     self.use_batch_norm = use_batch_norm
     ActorNetwork.__init__(self, sess, state_dim, action_dim, action_bound,
                           learning_rate, tau, batch_size)
예제 #2
0
 def __init__(self, sess, feature_number, state_dim, action_dim, action_bound, learning_rate,
              decay_rate, decay_steps, weight_decay, tau, batch_size,
              predictor_type, use_batch_norm,activation_function):
     self.feature_number = feature_number
     self.predictor_type = predictor_type
     self.use_batch_norm = use_batch_norm
     self.activation_function = activation_function
     self.weight_decay = weight_decay
     ActorNetwork.__init__(self, sess, state_dim, action_dim, action_bound, learning_rate,
                           decay_rate, decay_steps, tau, batch_size)
예제 #3
0
 def __init__(self,
              sess,
              state_dim,
              action_dim,
              action_bound,
              learning_rate,
              tau,
              batch_size,
              predictor_type,
              use_batch_norm,
              use_previous=False,
              auxiliary_prediction=0):
     self.predictor_type = predictor_type
     self.use_batch_norm = use_batch_norm
     self.use_previous = use_previous
     self.auxiliary_prediction = auxiliary_prediction
     ActorNetwork.__init__(self, sess, state_dim, action_dim, action_bound,
                           learning_rate, tau, batch_size)