squeezeDet icon indicating copy to clipboard operation
squeezeDet copied to clipboard

Published 20 hours ago • verified publisher icon BichenWuUCB

Interrupt and resume training from checkpoint

Open andreapiso opened this issue 8 years ago • 4 comments

Is it possible to interrupt the training and resume it from a specific checkpoint? My train dir and checkpoint dir are the same, but when I restarted the training it started back from step 0. Any suggestion on how to change the script to resume training from a checkpoint / freeze weights and create a .pb file from a checkpoint?

andreapiso avatar Mar 15 '17 02:03 andreapiso

I inserted this functionality and fixed a few bugs related to this and set up a model to train on VOC. Do you want a PR?

andreapiso avatar Mar 23 '17 00:03 andreapiso

@AndreaPisoni I haven't been able to restore a model checkpoint for training correctly either. Would you mind posting your solution?

RyanMorris95 avatar Mar 24 '17 03:03 RyanMorris95

@RyanMorris95 do you still need that? I cannot just post the code as I made a lot of changes but I can help... what have you tried so far?

andreapiso avatar Apr 03 '17 03:04 andreapiso

@AndreaPisoni I'd like to get some help on that, how can I contact you?

merryHunter avatar Sep 02 '17 14:09 merryHunter