I know model.train() is called at the start of every epoch. But can I call model.train() after every batch. My code is working when I called it after every batch.  I just wanted to know if there are any downsides to it.
Something like this:
for every epoch:
    for every batch:
         model.train()
         # some lines of training code