update config & readme
This commit is contained in:
parent
166b4162fb
commit
7f658d9672
@ -32,7 +32,7 @@ Your CSV file must contain the following columns:
|
||||
## 2. Config Preparation
|
||||
|
||||
|
||||
Please edit the correct data path and set your training parameters.
|
||||
Please edit the correct data path & pretrained model path and set your training parameters.
|
||||
|
||||
```yaml
|
||||
# Data configuration
|
||||
|
||||
@ -30,7 +30,7 @@ CSV文件必须包含以下列:
|
||||
|
||||
## 2. 准备config文件
|
||||
|
||||
data_path需要改成正确的数据路径,训练参数可以自己调节
|
||||
data_path及预训练模型路径需要修改,训练参数可以自己调节
|
||||
|
||||
```yaml
|
||||
# 数据配置
|
||||
|
||||
@ -2,7 +2,7 @@
|
||||
#这是一份模板config,用于kronos的csv自定义数据微调
|
||||
|
||||
data:
|
||||
data_path: "/xxxx/Kronos/finetune_csv2/data/HK_ali_09988_kline_5min_all.csv"
|
||||
data_path: "/xxxx/Kronos/finetune_csv/data/HK_ali_09988_kline_5min_all.csv"
|
||||
lookback_window: 512
|
||||
predict_window: 48
|
||||
max_context: 512
|
||||
@ -34,21 +34,21 @@ training:
|
||||
# model path configuration
|
||||
model_paths:
|
||||
# pretrained model path
|
||||
pretrained_tokenizer: "/mnt/DigitalHuman2D/boyuzhang/quant/Kronos/pretrained/Kronos-Tokenizer-base"
|
||||
pretrained_predictor: "/mnt/DigitalHuman2D/boyuzhang/quant/Kronos/pretrained/Kronos-base"
|
||||
pretrained_tokenizer: "/xxx/Kronos/pretrained/Kronos-Tokenizer-base"
|
||||
pretrained_predictor: "/xxx/Kronos/pretrained/Kronos-base"
|
||||
|
||||
# experiment name - other paths will be generated based on this
|
||||
exp_name: "HK_ali_09988_kline_5min_all"
|
||||
base_path: "/mnt/DigitalHuman2D/boyuzhang/quant/Kronos/finetune_csv/finetuned/"
|
||||
base_path: "/xxx/Kronos/finetune_csv/finetuned/"
|
||||
|
||||
# the following paths will be generated based on exp_name, no need to modify manually
|
||||
# way 1: leave empty string, the system will generate the full path
|
||||
base_save_path: "" # /xxxx/Kronos/finetune_csv/finetuned/{exp_name}
|
||||
finetuned_tokenizer: "" # /xxxx/quant/Kronos/finetune_csv/finetuned/{exp_name}/tokenizer/best_model
|
||||
finetuned_tokenizer: "" # /xxxx/Kronos/finetune_csv/finetuned/{exp_name}/tokenizer/best_model
|
||||
|
||||
# way 2: use template string, {exp_name} will be replaced with the actual experiment name
|
||||
# base_save_path: "/xxxx/Kronos/finetune_csv/finetuned/{exp_name}"
|
||||
# finetuned_tokenizer: "/xxxx/quant/Kronos/finetune_csv/finetuned/{exp_name}/tokenizer/best_model"
|
||||
# finetuned_tokenizer: "/xxxx/Kronos/finetune_csv/finetuned/{exp_name}/tokenizer/best_model"
|
||||
|
||||
tokenizer_save_name: "tokenizer"
|
||||
basemodel_save_name: "basemodel"
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user