MOM3

thingiverse

This doesn't appear to be a problem. You have shared Python code which is the beginning of some training and validation steps but there are numerous inconsistencies in your post that hinder any chance of reproducing and addressing what's being asked. It would appear as though you could take this process and simply insert a for loop inside of your function `run_epoch()` where `step=loop` if this is what your loop variable, also named step, stands for. ```python # run_epochs method definition here... def run_epochs(self): # Some code that defines your epoch. epochs = self.epochs # Assign some value to epochs loop = epochs + self.mini_epochs mini_epochs = epochs % self.max_size // self.batch_size step=loop for index, batch in enumerate(data_loader): ``` Or if it's actually a `while` loop instead: ```python # run_epochs method definition here... def run_epochs(self): # Some code that defines your epoch. epochs = self.epochs # Assign some value to epochs mini_epochs = epochs % self.max_size // self.batch_size step=loop for index, batch in enumerate(data_loader): ... ``` However please note that this isn't possible since the variables inside these loop would never actually equal.