roberta_des_512_4e4 / create_config.py
pere's picture
Saving weights and logs of step 10000
387b016
raw
history blame contribute delete
163 Bytes
from transformers import RobertaConfig
model_dir = "./" # ${MODEL_DIR}
config = RobertaConfig.from_pretrained("roberta-base")
config.save_pretrained(model_dir)