spatiotemporal_rwth_pre_training.yaml

# @package _global_

# to execute this experiment run:
# python train.py experiment=example

defaults:
  - spatiotemporal_pre_training
  - override /data: rwth_phoenix_2014

# all parameters below will be merged with parameters from default configurations set above
# this allows you to overwrite only specified parameters

tags: ["sign_language", "rwth_phoenix", "pre_training"]

data:
  batch_size: 8


trainer:
  min_epochs: 4
  max_epochs: 20 # more does not seem to improve loss.

model:
  warmup_step_interval: 150
  net:
    wav2vec2:
      _target_: src.models.components.spatiotemporal_encoder.SpatiotemporalEncoder.from_pretrained
      pretrained_model_name_or_path: "lukasbraach/spatiotemporal_bundestag_pretrain"