Issue : Inconsistent `model.train()` calls within the training loop

[PrinceSajjadHussain]

The code calls model.train() twice in the inner loop of the train function, which is redundant and can be removed. The second call to model.train() does not provide any additional value and might be confusing.