Skip to content

WIP: support using pre-trained weights#18

Open
leej3 wants to merge 4 commits intovfdev-5:object-detection-examplefrom
leej3:objdet_pt
Open

WIP: support using pre-trained weights#18
leej3 wants to merge 4 commits intovfdev-5:object-detection-examplefrom
leej3:objdet_pt

Conversation

@leej3
Copy link

@leej3 leej3 commented Dec 5, 2023

I have a couple of bugs to work through, from the ignite repo

docker run ... -v $PWD:/workspace/ignite pytorchignite/vision:latest /bin/bash
# and inside
pip install fire pycocotools
pip install ultralytics --no-dependencies
cd ignite/examples/object_detection
python main.py train --dataset=coco128 --data_path=coco128 --output_path=output --batch_size=64 --model=yolov8n-coco --with-clearml --num-workers=8 --num-epochs 200 --use-pt-weights

@leej3 leej3 changed the title support using pre-trained weights WIP: support using pre-trained weights Dec 5, 2023
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

Successfully merging this pull request may close these issues.

1 participant