mirror of
https://github.com/syssec-utd/pylingual.git
synced 2026-05-11 10:50:12 -07:00
Joel Flores
19 lines
507 B
JSON
19 lines
507 B
JSON
{
|
|
"base_repo_name": "sample_user/sample_segmenter_name",
|
|
"dataset_repo_name": "sample_user/segmentation-sample_dataset_name",
|
|
"pretrained_mlm_repo_name": "",
|
|
"cache_dir": "./cache-dir/",
|
|
"max_token_length": 512,
|
|
"dataset_percentage": 100,
|
|
"mlm_training_parameters": {
|
|
"batch_size": 48,
|
|
"epochs": 2,
|
|
"learning_rate": 5e-5
|
|
},
|
|
"segmentation_training_parameters": {
|
|
"batch_size": 48,
|
|
"epochs": 2,
|
|
"learning_rate": 2e-5
|
|
}
|
|
}
|