typo？ Is that should be train_net.py instead of train_det.py?

lucasjinreal / yolov7_d2

🔥🔥🔥🔥 (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! 🔥🔥🔥

GNU General Public License v3.0

3.13k stars 483 forks source link

Closed AllenPu closed 2 years ago

lucasjinreal commented 2 years ago

No, train_net is from d2 without any modification, it's standared training entrance in d2/

AllenPu commented 2 years ago

so for train a v7 model, which command we should run, python train_det.py ? Also, do you have the fine tune code for the pretrained model?