anticipation/anticipation/configs/long_term_anti/epic/verb_anti_vpretrain/gfb_gcn.py [5:27]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
model = dict(
    type='GFBModelGCN1',
    num_classes=num_classes,
    gfb_loss_weight=gfb_loss_weight,
    pre_trans=dict(
        type="mlp1D",
        num_layers=1,
        in_channels=2048,
        h_channels=2048,
        out_channels=2048
    ),
    gfb_module=dict(
        type="GCN",
        in_dim=2048,
        h_dim=2048,
        num_layers=1,
        dropout=0.7,
    ),
    backbone=None,
)

data = set_dataset_type(data, 'EpicFutureLabelsGFBAug')
work_dir = get_workdir(__file__)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



anticipation/anticipation/configs/long_term_anti/gtea/verb_anti_vpretrain/gfb_gcn.py [5:27]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
model = dict(
    type='GFBModelGCN1',
    num_classes=num_classes,
    gfb_loss_weight=gfb_loss_weight,
    pre_trans=dict(
        type="mlp1D",
        num_layers=1,
        in_channels=2048,
        h_channels=2048,
        out_channels=2048
    ),
    gfb_module=dict(
        type="GCN",
        in_dim=2048,
        h_dim=2048,
        num_layers=1,
        dropout=0.7,
    ),
    backbone=None,
)

data = set_dataset_type(data, 'EpicFutureLabelsGFBAug')
work_dir = get_workdir(__file__)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -