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

4 lines
89 B
Python

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