-
Notifications
You must be signed in to change notification settings - Fork 203
Open
Labels
bugSomething isn't workingSomething isn't working
Description
Is there an existing issue / discussion for this? | 是否已有关于该错误的issue或讨论?
- I have searched the existing issues / discussions | 我已经搜索过已有的issues和讨论
Is there an existing answer for this in tutorial? | 该问题是否在教程中有解答?
- I have searched tutorial | 我已经搜索过tutorial
Current Behavior | 当前行为
您好,感谢您提供的框架代码,我使用models中的DLinear模型在跑METR-LA数据集时,发现训练得到MAE/RMSE/MAPE都远低于一些Baselines的值
Expected Behavior | 期望行为
No response
Environment | 运行环境
- OS:Ubuntu 22.04.2
- DEVICE:NVIDIA GeForce RTX 5090
- NVIDIA Driver:580.76.05
- CUDA:13.0
- NVIDIA GPU Memory: 32GB
- PyTorch:2.8.0+cu128BasicTS logs | BasicTS日志
Steps To Reproduce | 复现方法
import os
from pathlib import Path
current_file = Path(__file__).resolve()
src_dir = current_file.parent.parent
if str(src_dir) not in sys.path:
sys.path.insert(0, str(src_dir))
project_root = src_dir.parent
os.chdir(project_root)
from basicts.models.DLinear import DLinear, DLinearConfig
from basicts.configs import BasicTSForecastingConfig
from basicts import BasicTSLauncher
def main():
dataset_name = 'METR-LA'
model_config = DLinearConfig(input_len=12,
output_len=12)
cfg = BasicTSForecastingConfig(
model=DLinear,
model_config = model_config,
dataset_name = dataset_name,
gpus = "0",
input_len = 12,
output_len = 12,
batch_size = 64,
use_timestamps = True,
)
BasicTSLauncher.launch_training(cfg)
if __name__ == "__main__":
main()
Anything else? | 备注
No response
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working