AniFileBERT / run_metadata.json
ModerRAS's picture
Improve anime filename parser model
e63569d
raw
history blame
600 Bytes
{
"experiment_name": "dmhy-char-full-relabel",
"data_file": "datasets/AnimeName/dmhy_weak_char.jsonl",
"tokenizer_variant": "char",
"vocab_file": "datasets/AnimeName/vocab.char.json",
"vocab_size": 6199,
"max_seq_length": 128,
"hidden_size": 256,
"num_hidden_layers": 4,
"num_attention_heads": 8,
"intermediate_size": 1024,
"train_samples": 619361,
"eval_samples": 12641,
"epochs": 2.0,
"batch_size": 256,
"learning_rate": 8e-05,
"warmup_steps": 300,
"seed": 48,
"device": "cuda",
"fp16": true,
"gradient_accumulation_steps": 1,
"dataloader_num_workers": 4
}