megvii-research / video_analyst

A series of basic algorithms that are useful for video understanding, including Single Object Tracking (SOT), Video Object Segmentation (VOS) and so on.
MIT License
832 stars 176 forks source link

save_final_model #119

Closed MARMOTatZJU closed 4 years ago

MARMOTatZJU commented 4 years ago

Save a final model (state_dict without optimizer state) after the last training epoch, inspired by the design of detectron2