mmpretrain/configs/repvgg/repvgg-B1g4_8xb32_in1k.py

4 lines
108 B
Python

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