mmpretrain/configs/_base_/models/vgg13bn.py

12 lines
296 B
Python

# model settings
model = dict(
type='ImageClassifier',
backbone=dict(
type='VGG', depth=13, norm_cfg=dict(type='BN'), num_classes=1000),
neck=None,
head=dict(
type='ClsHead',
loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
topk=(1, 5),
))