File size: 164 Bytes
84b5b20 |
1 2 3 4 5 6 7 8 |
from transformers import RobertaConfig
model_dir = "./" # ${MODEL_DIR}
config = RobertaConfig.from_pretrained("roberta-base")
config.save_pretrained(model_dir)
|
84b5b20 |
1 2 3 4 5 6 7 8 |
from transformers import RobertaConfig
model_dir = "./" # ${MODEL_DIR}
config = RobertaConfig.from_pretrained("roberta-base")
config.save_pretrained(model_dir)
|