32 lines
809 B
YAML
Executable File
32 lines
809 B
YAML
Executable File
model: 'MiniROADA'
|
|
data_name: 'EasyCom_ANTICIPATION'
|
|
task: 'ANTICIPATION'
|
|
loss: 'ANTICIPATION' # NONUNIFORM, originally
|
|
metric: 'AP' # AP or cAP
|
|
optimizer: 'AdamW'
|
|
device: 'cuda'
|
|
feature_pretrained: 'resnet50'
|
|
root_path: './data/EasyCom'
|
|
rgb_type: 'rgb_kinetics_resnet50'
|
|
flow_type: 'audio_wav2vec2'
|
|
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 # [background, target speaker speaking, other person speaking]
|
|
embedding_dim: 2048
|
|
hidden_dim: 1024
|
|
num_layers: 1
|
|
stride: 4
|
|
anticipation_length: 10
|
|
actionness: False
|
|
ignore_index: -100
|
|
seed: 20 |