update config & readme

This commit is contained in:
zhangboyu1 2025-10-09 18:14:04 +08:00
parent 166b4162fb
commit 7f658d9672
3 changed files with 8 additions and 8 deletions

View File

@ -32,7 +32,7 @@ Your CSV file must contain the following columns:
## 2. Config Preparation ## 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 ```yaml
# Data configuration # Data configuration

View File

@ -30,7 +30,7 @@ CSV文件必须包含以下列
## 2. 准备config文件 ## 2. 准备config文件
data_path需要改成正确的数据路径,训练参数可以自己调节 data_path及预训练模型路径需要改,训练参数可以自己调节
```yaml ```yaml
# 数据配置 # 数据配置

View File

@ -2,7 +2,7 @@
#这是一份模板config用于kronos的csv自定义数据微调 #这是一份模板config用于kronos的csv自定义数据微调
data: 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 lookback_window: 512
predict_window: 48 predict_window: 48
max_context: 512 max_context: 512
@ -34,21 +34,21 @@ training:
# model path configuration # model path configuration
model_paths: model_paths:
# pretrained model path # pretrained model path
pretrained_tokenizer: "/mnt/DigitalHuman2D/boyuzhang/quant/Kronos/pretrained/Kronos-Tokenizer-base" pretrained_tokenizer: "/xxx/Kronos/pretrained/Kronos-Tokenizer-base"
pretrained_predictor: "/mnt/DigitalHuman2D/boyuzhang/quant/Kronos/pretrained/Kronos-base" pretrained_predictor: "/xxx/Kronos/pretrained/Kronos-base"
# experiment name - other paths will be generated based on this # experiment name - other paths will be generated based on this
exp_name: "HK_ali_09988_kline_5min_all" 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 # 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 # way 1: leave empty string, the system will generate the full path
base_save_path: "" # /xxxx/Kronos/finetune_csv/finetuned/{exp_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
# way 2: use template string, {exp_name} will be replaced with the actual experiment name # 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}" # 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" tokenizer_save_name: "tokenizer"
basemodel_save_name: "basemodel" basemodel_save_name: "basemodel"