Is there a way to load weights from a particular epoch?

(Alex) #1

Hi.

I trained model on my own images and at the third epoch it hit 100% accuracy:
epochs
Is there a way to load weights from that particular epoch?

0 Likes

(Patrick Trampert) #2

You can add a Callback, e.g. like the following:

learner.callback_fns.append(partial(callbacks.SaveModelCallback, 
                              monitor='error_rate',
                              mode='min',
                              name='checkpoint'))
1 Like

(Alex) #3

Thanks! Exactly what I need.

0 Likes