Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

unable resume training #6

Open
chillum-codeX opened this issue Jun 28, 2022 · 1 comment
Open

unable resume training #6

chillum-codeX opened this issue Jun 28, 2022 · 1 comment

Comments

@chillum-codeX
Copy link

unable to resume training
when i pass the argument --resume it will add more epochs to my training

@chillum-codeX
Copy link
Author

python train.py --data pistols.yaml --cfg ./models_v5.0/yolov5s.yaml --weights zoo:cv/detection/yolov5-s/pytorch/ultralytics/coco/pruned_quant-aggressive_94?recipe_type=transfer --img 416 --batch-size 64 --hyp data/hyps/hyp.scratch.yaml --recipe ../recipes/yolov5.transfer_learn_pruned_quantized.md --optimizer SGD --device 0 --project yolov5-deepsparse --name yolov5s-sgd-pruned-quantized-transfer

this is the command i am using
demo one
when due to external factor training interrupt for ex power cut
when i try to resume the training it will add more epochs to my training
for ex. initial it was 220 epochs when i resume it was 440 epochs

i also used this one
python train.py --resume

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant