WarranWeng / ET-Net

Event-based Video Reconstruction Using Transformer, ICCV 2021.
49 stars 5 forks source link

Early stopping & checkpoint selection #3

Closed ercanburak closed 1 year ago

ercanburak commented 2 years ago

Hi, and thanks for sharing this.

Related to #2, I wonder how you performed model checkpoint selection to obtain the model file shared here and to get the quantitative results in the paper. Do you select the final model weights after the training is finished (after 300 epochs according to implementation details in the paper), or do you perform early stopping according to some criteria on a validation set, and select model weights from an earlier epoch?

WarranWeng commented 2 years ago

Thanks for your attention. Sorry for the late reply. I am busy with other things recently. We apply ealy stopping and choose the model with the best result in validation set.