LiteTCN / config.json
whenxuan's picture
whenxuan: update the model checkpoints
e7eaf5a
Raw
History Blame Contribute Delete
819 Bytes
{
"architectures": [
"LiteTCNModel"
],
"bucket_list": [
256,
512,
768,
1024,
1280,
1536,
1792,
2048,
2304,
2560,
2816,
3072
],
"casual_kernel_size": 3,
"channel_multiplier": [
2,
4,
4,
8
],
"context_length": 3072,
"dropout_rate": 0.1,
"dtype": "float32",
"final_channels": 512,
"hidden_size": 32,
"inception_kernel_sizes": [
3,
5,
11
],
"min_past": 64,
"model_type": "LiteTCNModel",
"num_hidden_layers": 1,
"prediction_length": 16,
"quantiles": [
0.05,
0.1,
0.2,
0.3,
0.4,
0.5,
0.6,
0.7,
0.8,
0.9,
0.95
],
"tie_word_embeddings": false,
"time_encoding_scale": 3072,
"transformers_version": "4.57.6",
"use_arcsinh": true,
"wide_kernel_size": 33
}