32 lines
777 B
YAML
32 lines
777 B
YAML
|
model: 'MiniROADA'
|
||
|
|
data_name: 'EGO4D_ANTICIPATION'
|
||
|
|
data_info_name: 'EGO4DSHUFFLE'
|
||
|
|
task: 'ANTICIPATION'
|
||
|
|
loss: 'ANTICIPATION' # NONUNIFORM, originally
|
||
|
|
metric: 'AP' # AP or cAP
|
||
|
|
optimizer: 'AdamW'
|
||
|
|
device: 'cuda'
|
||
|
|
feature_pretrained: 'resnet50'
|
||
|
|
root_path: './data/EGO4D'
|
||
|
|
rgb_type: 'rgb_kinetics_resnet50'
|
||
|
|
flow_type: 'audio_wav2vec2' # or 'none'?
|
||
|
|
annotation_type: 'target_perframe'
|
||
|
|
video_list_path: 'data_info/data_info.json'
|
||
|
|
output_path: './output_cfg'
|
||
|
|
window_size: 128
|
||
|
|
batch_size: 64
|
||
|
|
test_batch_size: 1
|
||
|
|
# loss_offset: 4 # loss weight a_t when L - {loss_offset} <= t <= L
|
||
|
|
num_epoch: 30
|
||
|
|
lr: 0.0001
|
||
|
|
weight_decay: 0.05
|
||
|
|
num_workers: 4
|
||
|
|
dropout: 0.20
|
||
|
|
num_classes: 3
|
||
|
|
embedding_dim: 2048
|
||
|
|
hidden_dim: 1024
|
||
|
|
num_layers: 1
|
||
|
|
stride: 4
|
||
|
|
anticipation_length: 10
|
||
|
|
actionness: False
|
||
|
|
ignore_index: -100
|