open-mmlab / mmrazor

OpenMMLab Model Compression Toolbox and Benchmark.
https://mmrazor.readthedocs.io/en/latest/
Apache License 2.0
1.48k stars 229 forks source link

How to train supernet on my own dataset? #236

Open JiuTMniJiaoXiaLuo opened 2 years ago

JiuTMniJiaoXiaLuo commented 2 years ago

Checklist

Describe the question you meet

Most of the existing datasets that can train supernet are ImageNet and CIFAR.What configs files should be modified in order to replace them with other datasets?

Post related information

f2118c77321dd07a0b744a79f677508

pprp commented 2 years ago

Hi, @JiuTMniJiaoXiaLuo , thank you for your interests.

As for your question, please refer to the mmcls document.

JiuTMniJiaoXiaLuo commented 2 years ago

Hi, @JiuTMniJiaoXiaLuo , thank you for your interests.

As for your question, please refer to the mmcls document.

Thank you for your reply,but after I created my dataset according to the steps, a new problem occurred. 1660743667127

pprp commented 2 years ago

Hi, @JiuTMniJiaoXiaLuo

You can refer to the similar issue: https://github.com/open-mmlab/mmtracking/issues/539

JiuTMniJiaoXiaLuo commented 2 years ago

Hi, @JiuTMniJiaoXiaLuo

You can refer to the similar issue: open-mmlab/mmtracking#539 Thank you for helping me, but I didn't understand. Which parameter did they delete and where did they delete it ?

pprp commented 2 years ago

Hi, @JiuTMniJiaoXiaLuo , I think they delete the file_client_args parameter.

Maybe the train_pipeline in your config is like this:

train_pipeline = [
    dict(type='LoadImageFromFile', file_client_args=file_client_args),
    ...
]

You can change to:

train_pipeline = [
    dict(type='LoadImageFromFile'),
    ...
]
JiuTMniJiaoXiaLuo commented 2 years ago

Hi, @JiuTMniJiaoXiaLuo , I think they delete the file_client_args parameter.

Maybe the train_pipeline in your config is like this:

train_pipeline = [
    dict(type='LoadImageFromFile', file_client_args=file_client_args),
    ...
]

You can change to:

train_pipeline = [
    dict(type='LoadImageFromFile'),
    ...
]

Thank you very much!