Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.utils.common.check_matching_networks

View source on GitHub

Check that two networks have matching input specs and variables.

tf_agents.utils.common.check_matching_networks(
    network_1, network_2
)

Args:

  • network_1: A network.Network.
  • network_2: A network.Network.

Raises:

  • ValueError: if the networks differ in input_spec, variables (number, dtype, or shape).
  • ValueError: if either of the networks has not been built yet (e.g. user must call create_variables).