mmpretrain/configs/repvgg/repvgg-B1_8xb32_in1k.py

4 lines
106 B
Python

_base_ = './repvgg-A0_8xb32_in1k.py'
model = dict(backbone=dict(arch='B1'), head=dict(in_channels=2048))