There is a max_steps in the config file

trainer:

benchmark: True

max_steps: 4000

Modify it to a higher value to prevent the training from stopping when it reached the limit