diff --git a/configs/_base_/recog_models/master.py b/configs/_base_/recog_models/master.py index 30e8241e..39eaef24 100644 --- a/configs/_base_/recog_models/master.py +++ b/configs/_base_/recog_models/master.py @@ -13,7 +13,7 @@ model = dict( cfg=dict( type='GCAModule', ratio=0.0625, - headers=1, + n_head=1, pooling_type='att', is_att_scale=False, fusion_type='channel_add'),