sql_env / configs /test_cpu.json
hjerpe's picture
Upload folder using huggingface_hub
9e64e71 verified
{
"model_name": "Qwen/Qwen3-0.6B",
"questions_path": "data/questions/questions_train.json",
"db_dir": "data/databases",
"output_dir": "/tmp/sqlenv_test",
"num_train_epochs": 1,
"max_steps": 2,
"per_device_train_batch_size": 2,
"num_generations": 2,
"max_completion_length": 256,
"step_budget": 3,
"logging_steps": 1,
"precision": "fp32",
"dataset_size": 4,
"enable_thinking": false,
"num_completions_to_print": 2
}