View source on GitHub |
Gets the total # of elements in the network's trainable variables.
tf_agents.bandits.policies.falcon_reward_prediction_policy.get_number_of_trainable_elements(
network: tf_agents.typing.types.Network
) -> tf_agents.typing.types.Float
Args | |
---|---|
network
|
A types.Network .
|
Returns | |
---|---|
The total number of elements in the network's trainable variables. |