How to store checkpoints periodically?

WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)

GNU General Public License v3.0

1.99k stars 520 forks source link

How to store checkpoints periodically? #137

Open neel04 opened 2 years ago

neel04 commented 2 years ago

Does anyone know how we can store the checkpoints periodically?

WongKinYiu commented 2 years ago

https://github.com/WongKinYiu/yolor/blob/main/train.py#L422

neel04 commented 2 years ago

@WongKinYiu Many Many thanks! 🤗 Just one last thing - can you tell me what's the output of the Yolo model when saving bounding boxers with the --save-text option? I want to convert them to absolute Pascal VOC coords.

I think the problem is mostly normalization - converting from [0,1] ==> [img_h, img_h] seems to be the troublesome part; I have square shaped images BTW.

Again, thanks for the quick reply!