diff --git a/finetune_csv/README.md b/finetune_csv/README.md index 4aa49ed..f9e58a9 100644 --- a/finetune_csv/README.md +++ b/finetune_csv/README.md @@ -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 diff --git a/finetune_csv/README_CN.md b/finetune_csv/README_CN.md index f72b99a..4f70653 100644 --- a/finetune_csv/README_CN.md +++ b/finetune_csv/README_CN.md @@ -30,7 +30,7 @@ CSV文件必须包含以下列: ## 2. 准备config文件 -data_path需要改成正确的数据路径,训练参数可以自己调节 +data_path及预训练模型路径需要修改,训练参数可以自己调节 ```yaml # 数据配置 diff --git a/finetune_csv/configs/config_ali09988_candle-5min.yaml b/finetune_csv/configs/config_ali09988_candle-5min.yaml index 94f4487..14774d2 100644 --- a/finetune_csv/configs/config_ali09988_candle-5min.yaml +++ b/finetune_csv/configs/config_ali09988_candle-5min.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"