_base_ = [
    '_base_/default_runtime.py', '_base_/schedules/schedule_1x_sgd.py',
    '_base_/models/fusion_4branch.py', '_base_/datasets/base_dataset.py'
]

load_from = 'pretrained/text0.7178_audio0.6702_video0.7139_image0.7039.pth'

model = dict(mode=3,
             head_config=dict(
                 fusion=dict(
                     _delete_=True,
                     type='ClsHead',
                     out_dim=82,
                     in_dim=1024,
                     loss=dict(type='MultiLabelBCEWithLogitsLoss'))
             ))
optimizer = dict(_delete_=True,
                 type='Adam',
                 lr=0.001,
                 weight_decay=0.0001)

