nip.trainers.rl_tensordict_base.update_schedule_iterator

nip.trainers.rl_tensordict_base.update_schedule_iterator(schedule: AgentUpdateSchedule)

Return a bool iterator which specifies on which iterations to update an agent.

Parameters:

schedule (AgentUpdateSchedule) – The update schedule.

Yields:

bool – Whether to update the agent on the current iteration.