tf_agents.utils.common.check_matching_networks

View source on GitHub

Check that two networks have matching input specs and variables.

network_1 A network.Network.
network_2 A network.Network.

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).