how to do resume of training? should i need to train for 300 epoch to get quantised ,model?

[akashAD98]

doing resume=True in train.py solved the problem

[akashAD98]

should i need to complete all 300 epoch training to get a quantized model? bcz I'm getting an error when i try to export 150 epoch model, & when i resume this training, it adds extra epochs.
here you can see i tried it for 300 epochs but when i stopped training & started resuming training its showed 389 epochs

[akashAD98]

getting this issue while converting to onnx, whats wrong here? should i need to do continuous training without stop?

[akashAD98]

even training is not completed ,only 2 epochs remaining ,its giving cuda out of memory ,

[dnth]

Hi @akashAD98

The quantization only happens at the last 2 epochs of the training. This is specified in the recipe file. So if you halt the training before the quantization epoch, you will not get a quantized model.

The screenshot below shows where you can change the quantization epoch.

① num_epochs is the total number of training epochs.

②quantization_start_epoch is the exact epoch where quantization begins.