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.99k stars 518 forks source link

colab train problem #4

Open tturkmen opened 3 years ago

tturkmen commented 3 years ago

!python train.py --data ./data/uav_coco.yaml --batch-size 8 --cfg ./models/uav_e6.yaml --name yolor-p6 --weights "" --hyp ./data/hyp.scratch.1280.yaml i have some problems Traceback (most recent call last): File "train.py", line 537, in train(hyp, opt, device, tb_writer, wandb) File "train.py", line 86, in train model = Darknet(opt.cfg).to(device) # create File "/content/yolor/models/models.py", line 530, in init self.module_defs = parse_model_cfg(cfg) File "/content/yolor/utils/parse_config.py", line 13, in parse_model_cfg with open(path, 'r') as f: FileNotFoundError: [Errno 2] No such file or directory: './models/uav_e6.yaml.cfg'

WongKinYiu commented 3 years ago

It seems you are using yaml style network config while the main branch only accept darknet style network config. Please try this branch https://github.com/WongKinYiu/yolor/tree/paper

Or if your are using darknet style network config, you have typos. --cfg ./models/uav_e6.yaml -> --cfg ./models/uav_e6.cfg