Add epoch to resume training

[kctezcan]

Is your feature request related to a problem? Please describe.

Now the code only takes the run_id to continue training from a previous training run_id (also see issue ). But I want to be able to specify from which epoch I want to resume training.

Describe the solution you'd like

I would like to add a new command line argument to specify this. Something like
parser.add_argument( "--epoch", type=int, default=None, help="From which epoch to continue training. Defaults to None.", )

Describe alternatives you've considered

No response

Additional context

No response

Organisation

No response