def __init__(self, **kwargs):
self.activations_factory = object_factory.ObjectFactory()
self.activations_factory.register_builder('relu', lambda **kwargs: tf.nn.relu)
self.activations_factory.register_builder('tanh', lambda **kwargs: tf.nn.tanh)
self.activations_factory.register_builder('sigmoid', lambda **kwargs: tf.nn.sigmoid)
self.activations_factory.register_builder('elu', lambda **kwargs: tf.nn.elu)
self.activations_factory.register_builder('selu', lambda **kwargs: tf.nn.selu)
self.activations_factory.register_builder('softplus', lambda **kwargs: tf.nn.softplus)
self.activations_factory.register_builder('None', lambda **kwargs: None)
self.init_factory = object_factory.ObjectFactory()
self.init_factory.register_builder('normc_initializer', lambda **kwargs: normc_initializer(**kwargs))
self.init_factory.register_builder('const_initializer', lambda **kwargs: tf.constant_initializer(**kwargs))
self.init_factory.register_builder('orthogonal_initializer',
lambda **kwargs: tf.orthogonal_initializer(**kwargs))
self.init_factory.register_builder('glorot_normal_initializer',
lambda **kwargs: tf.glorot_normal_initializer(**kwargs))
self.init_factory.register_builder('glorot_uniform_initializer',
lambda **kwargs: tf.glorot_uniform_initializer(**kwargs))
self.init_factory.register_builder('variance_scaling_initializer',
lambda **kwargs: tf.variance_scaling_initializer(**kwargs))
self.init_factory.register_builder('random_uniform_initializer',
lambda **kwargs: tf.random_uniform_initializer(**kwargs))
self.init_factory.register_builder('None', lambda **kwargs: None)
self.regularizer_factory = object_factory.ObjectFactory()
self.regularizer_factory.register_builder('l1_regularizer',
lambda **kwargs: tf.contrib.layers.l1_regularizer(**kwargs))
self.regularizer_factory.register_builder('l2_regularizer',
lambda **kwargs: tf.contrib.layers.l2_regularizer(**kwargs))
self.regularizer_factory.register_builder('l1l2_regularizer',
lambda **kwargs: tf.contrib.layers.l1l2_regularizer(**kwargs))
self.regularizer_factory.register_builder('None', lambda **kwargs: None)
def __init__(self):
self.model_factory = object_factory.ObjectFactory()
self.model_factory.register_builder(
'discrete_a2c', lambda network, **kwargs: models.ModelA2C(network))
self.model_factory.register_builder(
'discrete_a2c_lstm',
lambda network, **kwargs: models.LSTMModelA2C(network))
self.model_factory.register_builder(
'continuous_a2c',
lambda network, **kwargs: models.ModelA2CContinuous(network))
self.model_factory.register_builder(
'continuous_a2c_logstd',
lambda network, **kwargs: models.ModelA2CContinuousLogStd(network))
self.model_factory.register_builder(
'continuous_a2c_lstm',
lambda network, **kwargs: models.LSTMModelA2CContinuous(network))
self.model_factory.register_builder(
'continuous_a2c_lstm_logstd', lambda network, **kwargs: models.
LSTMModelA2CContinuousLogStd(network))
self.model_factory.register_builder(
'dqn', lambda network, **kwargs: models.AtariDQN(network))
self.network_factory = object_factory.ObjectFactory()
self.network_factory.register_builder(
'actor_critic', lambda **kwargs: network_builder.A2CBuilder())
self.network_factory.register_builder(
'dqn', lambda **kwargs: network_builder.DQNBuilder())
def __init__(self, **kwargs):
nn.Module.__init__(self, **kwargs)
self.activations_factory = object_factory.ObjectFactory()
self.activations_factory.register_builder(
'relu', lambda **kwargs: nn.ReLU(**kwargs))
self.activations_factory.register_builder(
'tanh', lambda **kwargs: nn.Tanh(**kwargs))
self.activations_factory.register_builder(
'sigmoid', lambda **kwargs: nn.Sigmoid(**kwargs))
self.activations_factory.register_builder(
'elu', lambda **kwargs: nn.ELU(**kwargs))
self.activations_factory.register_builder(
'selu', lambda **kwargs: nn.SELU(**kwargs))
self.activations_factory.register_builder(
'softplus', lambda **kwargs: nn.Softplus(**kwargs))
self.activations_factory.register_builder(
'None', lambda **kwargs: nn.Identity())
self.init_factory = object_factory.ObjectFactory()
#self.init_factory.register_builder('normc_initializer', lambda **kwargs : normc_initializer(**kwargs))
self.init_factory.register_builder(
'const_initializer', lambda **kwargs: _create_initializer(
nn.init.constant_, **kwargs))
self.init_factory.register_builder(
'orthogonal_initializer', lambda **kwargs: _create_initializer(
nn.init.orthogonal_, **kwargs))
self.init_factory.register_builder(
'glorot_normal_initializer', lambda **kwargs:
_create_initializer(nn.init.xavier_normal_, **kwargs))
self.init_factory.register_builder(
'glorot_uniform_initializer', lambda **kwargs:
_create_initializer(nn.init.xavier_uniform_, **kwargs))
self.init_factory.register_builder(
'variance_scaling_initializer',
lambda **kwargs: _create_initializer(
torch_ext.variance_scaling_initializer, **kwargs))
self.init_factory.register_builder(
'random_uniform_initializer', lambda **kwargs:
_create_initializer(nn.init.uniform_, **kwargs))
self.init_factory.register_builder(
'kaiming_normal', lambda **kwargs: _create_initializer(
nn.init.kaiming_normal_, **kwargs))
self.init_factory.register_builder('None', lambda **kwargs: None)