Changing epoch settings

[toastedbaguatte]

Hello there! I would like to modify the epochs for training and testing the data. However, I am not sure which part of the code actually affects the epoch.

[FateScript]

Take res18 version as example, plz change settings here.