Upload model checkpoint: clf-imdb-pythia-410m-s-0-lat-tr-rt-t-0-adv-training-round-1 (main)
016d326
verified
{ | |
"_name_or_path": "AlignmentResearch/robust_llm_pythia-410m_clf_imdb_v-ian-079_s-0", | |
"architectures": [ | |
"GPTNeoXForSequenceClassification" | |
], | |
"attention_bias": true, | |
"attention_dropout": 0.0, | |
"bos_token_id": 0, | |
"classifier_dropout": 0.1, | |
"eos_token_id": 0, | |
"hidden_act": "gelu", | |
"hidden_dropout": 0.0, | |
"hidden_size": 1024, | |
"initializer_range": 0.02, | |
"intermediate_size": 4096, | |
"layer_norm_eps": 1e-05, | |
"max_position_embeddings": 2048, | |
"model_type": "gpt_neox", | |
"num_attention_heads": 16, | |
"num_hidden_layers": 24, | |
"pad_token_id": 0, | |
"partial_rotary_factor": 0.25, | |
"problem_type": "single_label_classification", | |
"rope_scaling": null, | |
"rope_theta": 10000, | |
"rotary_emb_base": 10000, | |
"rotary_pct": 0.25, | |
"tie_word_embeddings": false, | |
"torch_dtype": "float32", | |
"transformers_version": "4.46.2", | |
"use_cache": false, | |
"use_parallel_residual": true, | |
"vocab_size": 50304 | |
} | |