mongolian-roberta-base / run_config.py
bayartsogt's picture
Saving weights and logs of epoch 1
84b5b20
raw
history blame
164 Bytes
from transformers import RobertaConfig
model_dir = "./" # ${MODEL_DIR}
config = RobertaConfig.from_pretrained("roberta-base")
config.save_pretrained(model_dir)