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

想用ResNet18,请问怎么配置 #49

Open
Dominix-411 opened this issue Sep 17, 2020 · 0 comments
Open

想用ResNet18,请问怎么配置 #49

Dominix-411 opened this issue Sep 17, 2020 · 0 comments

Comments

@Dominix-411
Copy link

大佬。我的数据集是特征比较少的图片,想用resnet18。
请问configs里
# model settings model = dict( type='PolarMask', pretrained='open-mmlab://resnet50_caffe', backbone=dict( type='ResNet', depth=50, num_stages=4, out_indices=(0, 1, 2, 3), frozen_stages=1, norm_cfg=dict(type='BN', requires_grad=False), style='caffe'),
pretrained='open-mmlab://resnet50_caffe',要改成pretrained='open-mmlab://resnet18_caffe'吗。好像没有这个。
要是光改depth,会有RuntimeError: Given groups=1, weight of size 256 512 1 1, expected input[4, 128, 64, 64] to have 512 channels, but got 128 channels instead这个错。实在是迷茫了。望大佬解答,谢谢

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

1 participant