How can I resume with recovery weights?

[ohwi]

Hi. I'm looking for the way to resume with recovery weights, starting in the middle of epoch.

However, I could not find the way to do this.

How can I do this?

Thank you.

[rwightman]

@ohwi there is no support for resuming in the middle of the epoch w/ the recovery, you'd have to resume at the start of the epoch you were in and it'd basically be equivalent to extending that epoch with repeated samples. If you reallly really wanted to do it, you'd have to parse out the batch idx from recovery filename and hack the train loop to start from there the first pass.

[ohwi]

Thank you for you advice! I thought that the recovery is for starting in the middle of the epoch.

As you mentioned, hacking the first train loop without loading images will be the great solution. I'm going to try this. Thank you!!