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
|
## 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
|
||||||
|
|||||||
@ -30,7 +30,7 @@ CSV文件必须包含以下列:
|
|||||||
|
|
||||||
## 2. 准备config文件
|
## 2. 准备config文件
|
||||||
|
|
||||||
data_path需要改成正确的数据路径,训练参数可以自己调节
|
data_path及预训练模型路径需要修改,训练参数可以自己调节
|
||||||
|
|
||||||
```yaml
|
```yaml
|
||||||
# 数据配置
|
# 数据配置
|
||||||
|
|||||||
@ -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"
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user