tf_agents.policies.utils.get_model_index

Returns the model index for a specific arm.

The number of models depends on the observation format: If the policy accepts per-arm features, there is only one single model used for every arm. Otherwise there is a model for every arm.

arm_index The index of the arm for which the model index is needed.
accepts_per_arm_features (bool) Whether the policy works with per-arm features.

The index of the model for the arm requested.