search

open-mmlab / mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark
https://mmpretrain.readthedocs.io/en/latest/
Apache License 2.0
3.46k stars 1.07k forks source link

Fix deploy error #1894

Open Y-T-G opened 6 months ago

Y-T-G commented 6 months ago

Motivation

Deployment fails for LeViT.

Modification

Fix error in self.switch_to_deploy()

CLAassistant commented 6 months ago

CLA assistant check
All committers have signed the CLA.