simpledet icon indicating copy to clipboard operation
simpledet copied to clipboard

How to save the weight file every two epochs

Open dongzhenguo2016 opened this issue 5 years ago • 1 comments

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

dongzhenguo2016 avatar Jun 19 '20 03:06 dongzhenguo2016

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

zehuichen123 avatar Jun 19 '20 11:06 zehuichen123