Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] 请问如何对这些模型进行微调,说明文档里并未找到 #1937

Open
1 task
abcc235 opened this issue Aug 27, 2024 · 1 comment
Open
1 task

Comments

@abcc235
Copy link

abcc235 commented Aug 27, 2024

描述该功能

1.希望可以整理如何对这些模型进行微调的文档

是否希望自己实现该功能?

  • 我希望自己来实现这一功能,并向 MMPreTrain 贡献代码!
@bear-coder-9527
Copy link

check this:
https://mmpretrain.readthedocs.io/zh-cn/latest/notes/finetune_custom_dataset.html

base = [
'../base/models/resnet50.py', # 模型设置
'../base/datasets/imagenet_bs32.py', # 数据设置
'../base/schedules/imagenet_bs256.py', # 训练策略设置
'../base/default_runtime.py', # 运行设置
]

>>>>>>>>>>>>>>> 在这里重载模型相关配置 >>>>>>>>>>>>>>>>>>>

model = dict(
backbone=dict(
frozen_stages=2,
init_cfg=dict(
type='Pretrained',
checkpoint='https://download.openmmlab.com/mmclassification/v0/resnet/resnet50_8xb32_in1k_20210831-ea4938fc.pth',
prefix='backbone',
)),
head=dict(num_classes=10),
)

<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants