How to reset my early stop patience counter when scheduler step() changed the learning rate?

2

The number of “bad epochs” will already be reset as seen here so if you haven’t changed the cooldown the scheduler should again apply its patience schedule.