Skip to content

用自己的数据集来 预训练 模型时,如何配置 #59

@emanlee

Description

@emanlee

作者您好!

我们想用自己的数据集来 预训练 模型。
我们看了一下 TSFormer_METR-LA.py 文件,里面有些配置不明白。因此,请教一下。

CFG.DATASET_INPUT_LEN = 288 * 7 这个288和7分别表示什么,谢谢!

CFG.MODEL.PARAM = {
"patch_size":12, ############ 请问这个12表示输出时间步吗?
"in_channel":1,
"embed_dim":96,
"num_heads":4,
"mlp_ratio":4,
"dropout":0.1,
"num_token":288 * 7 / 12, ############ 请问这个地方为什么除以12?
"mask_ratio":0.75,
"encoder_depth":4,
"decoder_depth":1,
"mode":"pre-train"
}

从 TSFormer_METR-LA.py 文件看不出原始的输入数据的文件名以及文件扩展名,请问输入数据应该放到哪个文件夹,并且,文件取名有什么要求吗? 是不是要这样取名 METR-LA.h5 ? 还是要类似于 scaler_in2016_out12.pkl ?

谢谢!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions