请问这个框架不支持蒸馏吗？

[ersanliqiao]

Traceback (most recent call last): File "/opt/conda/lib/python3.8/runpy.py", line 194, in _run_module_as_main return _run_code(code, main_globals, None, File "/opt/conda/lib/python3.8/runpy.py", line 87, in _run_code exec(code, run_globals) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/main.py", line 31, in main() File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/main.py", line 25, in main args.run(args) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/commands/train.py", line 161, in _main launch(main, args.num_gpus_per_machine, args.num_machines, args=args, start_method=args.fork_method) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/utils/env/launch.py", line 68, in launch main_func(*(args,)) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/commands/train.py", line 140, in main runner = RUNNER_REGISTRY.get(runner_cfg['type'])(cfg, **runner_cfg['kwargs']) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/utils/general/registry.py", line 81, in get assert module_name in self, '{} is not supported, avaiables are:{}'.format(module_name, self) AssertionError: kd is not supported, avaiables are:{'base': <class 'up.runner.base_runner.BaseRunner'>}

[yqyao]

If you want to use distill, you need add tasks export DEFAULT_TASKS=cls:distill @ersanliqiao

[ersanliqiao]

If you want to use distill, you need add tasks export DEFAULT_TASKS=cls:distill @ersanliqiao

多谢，改完可以跑了，另外，想问下我跑nas，导入task也改了（export DEFAULT_TASKS=cls:bignas ），但是还是报错： no custom work to be done.

• export DEFAULT_TASKS=cls:bignas
• DEFAULT_TASKS=cls:bignas
• bash scripts/dist_train.sh 1 /data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/configs/nas/bignas/cls/bignas_regnet_800m_train_supnet.yaml No module named 'petrel_client' init petrel failed No module named 'spring_aux' 2022-12-06 17:40:51,629-rk0-normalize.py#37:import error No module named 'mqbench'; If you need Mqbench to quantize model, you should add Mqbench to this project. Or just ignore this error. 2022-12-06 17:40:51,629-rk0-normalize.py#44:import error No module named 'msbench'; If you need Msbench to prune model, you should add Msbench to this project. Or just ignore this error. Traceback (most recent call last): File "/opt/conda/lib/python3.8/runpy.py", line 194, in _run_module_as_main return _run_code(code, main_globals, None, File "/opt/conda/lib/python3.8/runpy.py", line 87, in _run_code exec(code, run_globals) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/main.py", line 31, in main() File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/main.py", line 25, in main args.run(args) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/commands/train.py", line 161, in _main launch(main, args.num_gpus_per_machine, args.num_machines, args=args, start_method=args.fork_method) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/utils/env/launch.py", line 68, in launch main_func(*(args,)) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/commands/train.py", line 140, in main runner = RUNNER_REGISTRY.get(runner_cfg['type'])(cfg, **runner_cfg['kwargs']) File "/data/juicefs_hz_cv_v3/11105507/AAAI2022/united-perception2/up/utils/general/registry.py", line 81, in get assert module_name in self, '{} is not supported, avaiables are:{}'.format(module_name, self) AssertionError: bignas is not supported, avaiables are:{'base': <class 'up.runner.base_runner.BaseRunner'>} train <class 'up.commands.train.Train'>

[yqyao]

Maybe you need to set DEFAULT_TASKS=cls:nas @ersanliqiao

[TheWangYang]

Why did I add export DEFAULT_TASKS=cls:distill as you mentioned above, but I got the following error:

ImportError: cannot import name 'gpu_iou_overlap' from 'up.extensions' (/ssd3/wyy/projects/United-Perception/up/extensions/__init__.py)

This seems to be the same as the error I reported before when I did not export any tasks. How can I solve it? Looking forward to your reply? @yqyao

[yqyao]

If you want to use advanced features, maybe you need to compile extensions. following the docs @TheWangYang