File size: 295 Bytes
ce41b9a |
1 2 3 4 5 6 7 8 9 10 |
'''Create the configuration for the model'''
from transformers import RobertaConfig
from .utils import model_dir
# Currently it merely copies the `roberta-base` config, but we can change this
# of course
config = RobertaConfig.from_pretrained("roberta-base")
config.save_pretrained(model_dir)
|