| { | |
| "backbone": "NeoQuasar/Kronos-base", | |
| "tokenizer_id": "NeoQuasar/Kronos-Tokenizer-base", | |
| "device": "cuda", | |
| "dtype": "torch.bfloat16", | |
| "window_size": 60, | |
| "forecast_horizon": 15, | |
| "data_clip": 5.0, | |
| "batch_size": 256, | |
| "grad_accum_steps": 1, | |
| "epochs": 20, | |
| "learning_rate": 2e-05, | |
| "weight_decay": 0.01, | |
| "lr_scheduler": "cosine", | |
| "warmup_steps": 500, | |
| "gradient_clip": 1.0, | |
| "freeze_tokenizer": true, | |
| "freeze_bottom_layers": 6, | |
| "num_workers": 2, | |
| "n_train_samples": 200000, | |
| "n_val_samples": 40000, | |
| "early_stopping_patience": 3, | |
| "save_every_n_epochs": 5, | |
| "output_dir": "/content/drive/MyDrive/TradeIntelix/kronos_finetune/checkpoints", | |
| "use_torch_compile": false, | |
| "compile_mode": "max-autotune" | |
| } |