search

tusen-ai / simpledet

A Simple and Versatile Framework for Object Detection and Instance Recognition
Apache License 2.0
3.08k stars 486 forks source link

How to save the weight file every two epochs #338

Open dongzhenguo2016 opened 4 years ago

dongzhenguo2016 commented 4 years ago

Simpledet saves a weight file once for each epoch during training. Since the epoch I set is relatively large, now I want to save the weight file every two epochs, where do I need to set it? Thank you

zehuichen123 commented 4 years ago

maybe you can add a if-else clause here to decide whether to save it or not :) https://github.com/TuSimple/simpledet/blob/bcccd77d5c763901a90afe4bb503849bb7ee4dbc/core/detection_module.py#L185