Is there a way to inform the agent that a new episode starts when defining trainer using build_trainer?
trainer
build_trainer