def test_check_no_shared_variables_expect_fail(self):
dense_layer = tf.keras.layers.Dense(3)
q_net_1 = networks_test_utils.KerasLayersNet(self._observation_spec,
self._action_spec,
dense_layer)
q_net_2 = networks_test_utils.KerasLayersNet(self._observation_spec,
self._action_spec,
dense_layer)
with self.assertRaises(ValueError):
common.check_no_shared_variables(q_net_1, q_net_2)
def test_check_no_shared_variables(self):
layer_1 = tf.keras.layers.Dense(3)
layer_2 = tf.keras.layers.Dense(3)
q_net_1 = networks_test_utils.KerasLayersNet(self._observation_spec,
self._action_spec,
layer_1)
q_net_2 = networks_test_utils.KerasLayersNet(self._observation_spec,
self._action_spec,
layer_2)
common.check_no_shared_variables(q_net_1, q_net_2)