search

WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

Load weights error #63

Open APeiZou opened 2 years ago

APeiZou commented 2 years ago

when I training the model with following script with paper branch. python train.py --batch-size 32 --img 640 640 --data data/youcha.yaml --cfg cfg/yolor_w6.cfg --weights yolor-w6.pt --device 1 --name yolor-w6 --hyp hyp.scratch.1280.yaml --epochs 120 An error occurs image

APeiZou commented 2 years ago

@WongKinYiu

WongKinYiu commented 2 years ago

https://github.com/WongKinYiu/yolor/tree/paper#training