Check point and Fine Tune in C3W2_Assignment

Hey there,

why don’t we restore detection_model._box_predictor._prediction_heads ?

This my fool question, because we will fine tune the XXXX ClassHead/ClassPredictor, and it is now confusing me.

The same doesn’t happen on “box”, we restore then and fine tune associated variables.

Can anyone give answer intuitively or any text which can clarify this?

Thank you.