Replies: 1 comment
-
|
@matbord that should be fine, but FYI if you are simply loading a pretrained model and not modifying it's structure there is no need to pass a --cfg argument, a --weights argument will suffice: |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hi,
I would like to train a model, save it and then use it to start another training. I want to do this because I don't have enough memory to fit all my dataset in one training.
I would like to know if this procedure is correct before starting.
if first model :
!python train.py --img 1316 --batch 4 --epochs 20 --data ../path_to_data1 --cfg ./models/yolov5x.yaml --weights yolov5x.pt --name yolov5x_model1--cacheelse:
!python train.py --img 1316 --batch 4 --epochs 20 --data../path_to_data2 --cfg ./models/yolov5x.yaml --weights yolov5x_model1.pt --name yolov5x_model2 --cacheBeta Was this translation helpful? Give feedback.
All reactions