I am working on a personal project, that should be deterministic in nature.
After each batch during training, model will go to next state like how finite state machin