config.yaml

model:
  name: "voxmenthe/modernbert-imdb-sentiment"
  output_dir: "checkpoints"
  max_length: 880 # 256
  dropout: 0.1
  pooling_strategy: "mean" # Current default, change as needed

inference:
  # Default path, can be overridden
  model_path: "checkpoints/mean_epoch5_0.9575acc_0.9575f1.pt" 
  # Using the same max_length as training for consistency
  max_length: 880 # 256