autotimes_config.json

{
    "model_name": "ETH_5m_tempnorm",
    "root_path": "/content/drive/MyDrive/eth_autotimes/",
    "data_path": "eth_5m.csv",
    "llm_type": "llama",
    "model_id": "eth_5m",
    "model": "AutoTimes_Llama",
    "data": "ETH",
    "features": null,
    "seq_len": 480,
    "label_len": 384,
    "token_len": 96,
    "tokenize_marks": [
        true
    ],
    "pred_len": 96,
    "minutes": [
        5
    ],
    "test_seq_len": 480,
    "test_label_len": 384,
    "test_pred_len": 96,
    "batch_size": 64,
    "learning_rate": 0.0005,
    "weight_decay": 0,
    "train_epochs": 10,
    "tmax": 10,
    "gpu": 0,
    "use_amp": true,
    "cosine": true,
    "val_set_shuffle": false,
    "drop_last": false,
    "num_workers": 4,
    "checkpoints": "./checkpoints",
    "test_dir": "./test",
    "test_file_name": "best_model.pth",
    "visualize": false,
    "loss": "MSE",
    "use_multi_gpu": false,
    "local_rank": null,
    "seasonal_patterns": null,
    "frequency_map": null,
    "des": "test",
    "lradj": "type1",
    "patience": 3,
    "dropout": 0.1,
    "llm_ckp_dir": "./llama",
    "mlp_hidden_dim": 512,
    "mlp_hidden_layers": 4,
    "mlp_activation": "gelu",
    "mix_embeds": true,
    "scale": true,
    "train_fraction": 0.7,
    "test_fraction": 0.25,
    "scaler_type": "standard",
    "extra_kwargs": {}
}