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

    for i in range(num_modes):
      for j in range(i+1, num_modes):
        modes_differ = np.linalg.norm(modes[i] - modes[j]) > 1e-6
        if modes_differ:
          break

    self.assertEqual(training, modes_differ)


if __name__ == "__main__":

Instances