Comments (3)
你好,我想问一下,如果我想使用加入注意力机制的Yolov4,使用了yolov4的预训练文件,但是注意力模块并没有预训练参数,我可以在加载yolov4预训练文件之后便把这部分参数冻结,然后用自定义数据集只训练注意力模块中参数,训练完成后解除参数冻结,再用自己的数据集训练整个模型,这个操作可行吗?因为在imagenet上预训练这个模型,没有好的GPU资源。。。
一般来说是先冻结整个backbone的参数,训练其他参数,然后在finetune整个模型。我不清楚你数据集学习的难度,不过你的想法可以试一下。
from yolov4-pytorch.
你好,我想问一下,如果我想使用加入注意力机制的Yolov4,使用了yolov4的预训练文件,但是注意力模块并没有预训练参数,我可以在加载yolov4预训练文件之后便把这部分参数冻结,然后用自定义数据集只训练注意力模块中参数,训练完成后解除参数冻结,再用自己的数据集训练整个模型,这个操作可行吗?因为在imagenet上预训练这个模型,没有好的GPU资源。。。
一般来说是先冻结整个backbone的参数,训练其他参数,然后在finetune整个模型。我不清楚你数据集学习的难度,不过你的想法可以试一下。
作者您好 可以问您个吗? 为什么我使用了注意力机制 反而比原模型稍微降低了一些,不知道怎么回事。需要注意什么吗?谢谢
from yolov4-pytorch.
你好,我想问一下,如果我想使用加入注意力机制的Yolov4,使用了yolov4的预训练文件,但是注意力模块并没有预训练参数,我可以在加载yolov4预训练文件之后便把这部分参数冻结,然后用自定义数据集只训练注意力模块中参数,训练完成后解除参数冻结,再用自己的数据集训练整个模型,这个操作可行吗?因为在imagenet上预训练这个模型,没有好的GPU资源。。。
如果你加入了注意力机制,原预训练权重就不能再用了。
from yolov4-pytorch.
Related Issues (20)
- 问题 HOT 1
- RuntimeError: shape '[256, 256, 3, 3]' is invalid for input of size 16301 HOT 3
- 如何修改训练图片大小并且使用多尺度训练策略 HOT 2
- 关于CBAM的问题 HOT 2
- ValueError: not enough values to unpack (expected 3, got 0) HOT 2
- ValueError: not enough values to unpack (expected 3, got 2) HOT 1
- 可以放入负样本嘛 HOT 1
- CUDA_VISIBLE_DEVICES=0 nohup python -u train.py --weight_path weight/yolov4.weights --gpu_id 0 > nohup.log 2>&1 & HOT 2
- 训练好的模型百度网盘提取码是多少? HOT 1
- How to train model using dataset with another resolution?
- When I set the activate='relu', I met the RuntimeError. (Solved) HOT 1
- please provide best weights after training HOT 1
- Darknet pretrained weight file link corrupted HOT 1
- YOLOv4 with CBAM HOT 1
- Evaluator.py map issue with dataset HOT 5
- Error for detection HOT 1
- Error in detection HOT 2
- 多尺度训练,NUMBER_WORKERS
- Pretrained mobilev2 not accessible
- 姐姐,使用voc_best.pt导入失败呀
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from yolov4-pytorch.