simpledet
simpledet copied to clipboard
How to save the weight file every two epochs
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
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