Skip to content

Commit

Permalink
Merge pull request #100 from Marco-Nguyen/patch-2
Browse files Browse the repository at this point in the history 
Update README.md
  • Loading branch information
@Bobo-y
Bobo-y committed Jun 23, 2022
2 parents df06e69 + db809f2 commit 6782c9d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ For training and Testing, it's same like yolov5.
2. check out configs/model_*.yaml, choose backbone. and change nc to your dataset. please refer support_backbone in models.backbone.__init__.py
3.
```shell script
$ python scripts/train.py --batch 16 --epochs 5 --data configs/data.yaml --cfg confgis/model_XXX.yaml
$ python scripts/train.py --batch 16 --epochs 5 --data configs/data.yaml --cfg configs/model_XXX.yaml
```

A google colab demo in train_demo.ipynb
Expand Down

2 comments on commit 6782c9d

@ratom
Copy link

@ratom ratom commented on 6782c9d Jun 30, 2022 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

image
I got this error when I tried to train my model

@Bobo-y
Copy link
Owner Author

@Bobo-y Bobo-y commented on 6782c9d Jun 30, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@ratom downgrade your PyYaml to 5.XXX

Please sign in to comment.