Adapters
SkyReach_Companion_AI / train_config.json
SkyXIntl's picture
Create train_config.json
8359372 verified
raw
history blame contribute delete
302 Bytes
{
"model_name": "bert-base-uncased",
"training_args": {
"output_dir": "./results",
"num_train_epochs": 3,
"per_device_train_batch_size": 8,
"per_device_eval_batch_size": 8,
"warmup_steps": 500,
"weight_decay": 0.01,
"logging_dir": "./logs",
"logging_steps": 10
}
}