Spaces:
Running
Running
File size: 683 Bytes
41b743c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# Config parameters for router data construction:
api_key: ""
llm_description_path: 'configs/new_LLM_Descriptions_with_think.json'
llm_embedding_path: 'configs/new_llm_description_embedding_with_think.pkl'
train_data_path: 'data/train.csv'
test_data_path: 'data/test.csv'
query_response_length: 512
# Config parameters for graph_router training:
seed: 2
wandb_key: ''
model_path: 'model_path/all_performance_best_model.pth'
train_epoch: 1000
scenario: "Performance First" # "Performance First", "Balance", and "Cost First"
llm_num: 40
learning_rate: 0.0001
weight_decay: 0.0001
train_mask_rate: 0.5
batch_size: 64
num_task: 15
split_ratio: [0.9, 0.1]
embedding_dim: 8
edge_dim: 3
|