-
Notifications
You must be signed in to change notification settings - Fork 45
Open
Description
作者您好!
我们想用自己的数据集来 预训练 模型。
我们看了一下 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
Labels
No labels