Media-Smart / vedadet

A single stage object detection toolbox based on PyTorch
Apache License 2.0
498 stars 128 forks source link

您好,我自己想训练自己的数据集,但是一直加载有问题,请问一下我不想加载使用resnet50的预训练模型,我该更改什么地方呢,谢谢您 #35

Closed juliusyang97 closed 3 years ago

juliusyang97 commented 3 years ago

21 这个问题一直让我无法进行下去

我想着只加载resnet50网络结构,不加载官方的预训练模型,这种方法可行吗; 我查了pytorch加载预训练模型的方法,但是我没有在本项目中找到如何更改

mike112223 commented 3 years ago

If you do not want to use pretrained model, you can simply delete the "weights" in config.