pretrained_convnext-v2-base_32xb32_in1k-384px.py

_base_ = ['../gender_detection/convnext-v2-base_32xb32_in1k-384px.py']


# >>>>>>>>>>>>>>> Override model settings here >>>>>>>>>>>>>>>>>>>
model = dict(
    backbone=dict(
        # frozen_stages=8,
        init_cfg=dict(
            type='Pretrained',
            checkpoint='../pre-trained-models/convnext-v2-base_fcmae-in21k-pre_3rdparty_in1k-384px_20230104-379425cc.pth',
            prefix='backbone',
        ))
)

visualizer = dict(
    type='UniversalVisualizer', 
    vis_backends=[
        dict(type='TensorboardVisBackend'), 
        dict(type='LocalVisBackend'),
        dict(
            type='WandbVisBackend',
            init_kwargs=dict(project='gender_detection'),
            define_metric_cfg=[
                dict(name='accuracy/top1', step_metric='epoch'),
                dict(name='accuracy/top2', step_metric='epoch'),
            ]
        )])

custom_hooks = [
    dict(
        type='EarlyStoppingHook',
        monitor='accuracy/top1',
        patience=100,
        stopping_threshold=99.0
        
    )]