9a3e9d21273da7ae40da9f70cb6df1b077b08105,tf_agents/networks/actor_distribution_network_test.py,ActorDistributionNetworkTest,testDropoutFCLayersWithConv,#ActorDistributionNetworkTest#Any#,117

Before Change

      ("TrainingTrue", True,),
      ("TrainingFalse", False))
  def testDropoutFCLayersWithConv(self, training):
    tf.compat.v1.set_random_seed(0)
    observation_spec = tensor_spec.BoundedTensorSpec((8, 8, 3), tf.float32, 0,
                                                     1)
    time_step_spec = ts.time_step_spec(observation_spec)
    time_step = tensor_spec.sample_spec_nest(time_step_spec, outer_dims=(1,))

After Change

        fc_layer_params=[5],
        dropout_layer_params=[0.5])

    modes = []
    num_modes = 10
    for _ in range(num_modes):
      action_distributions, _ = net(
          time_step.observation, time_step.step_type, (), training=training)
      modes.append(action_distributions.mode())

    self.evaluate(tf.compat.v1.global_variables_initializer())
    modes = self.evaluate(modes)

    modes_differ = False
    for i in range(num_modes):
      for j in range(i+1, num_modes):

Instances