mmpretrain/configs/riformer/deploy/riformer-m48-deploy_8xb64_i...

4 lines
83 B
Python

_base_ = '../riformer-m48_8xb64_in1k.py'
model = dict(backbone=dict(deploy=True))