Madjakul/deep-stylometry-modernbert-layerwise
0
1# test_layerwise.yml — Evaluation with layerwise pooling checkpoint2 3project_name: "deep-stylometry"4 5model:6 pooling_method: "layerwise"7 skip_list: false8 mean_center: false9 10data:11 ds_name: "halvest"12 batch_size: 6413 tokenizer_name: "answerdotai/ModernBERT-base"14 max_length: 51215 padding: "do_not_pad"16 truncation: "longest_first"17 add_special_tokens: true18 map_batch_size: 100019 load_from_cache_file: true20 subsets: ["base-2", "base-4", "base-6", "base-8", "base-10"]21 shuffle: true22 test_subset: "base-2"23 24test:25 device: "gpu"26 num_devices: 127 log_every_n_steps: 128 precision: "16-mixed"29 use_wandb: true30 