tanmayshishodia
32 lines
957 B
Bash
32 lines
957 B
Bash
#!/bin/bash
|
|
|
|
# Script to finetune a model with specific configurations
|
|
# Adjust the parameters below as needed. For a full list of options and descriptions, run the script with the --help flag.
|
|
|
|
export TF_CPP_MIN_LOG_LEVEL=2 XLA_PYTHON_CLIENT_PREALLOCATE=false
|
|
|
|
python3 finetune.py \
|
|
--model-name="google/timesfm-1.0-200m" \
|
|
--backend="cpu" \
|
|
--horizon-len=128 \
|
|
--context-len=512 \
|
|
--freq="15min" \
|
|
--data-path="../datasets/ETT-small/ETTm1.csv" \
|
|
--num-epochs=100 \
|
|
--learning-rate=1e-3 \
|
|
--adam-epsilon=1e-7 \
|
|
--adam-clip-threshold=1e2 \
|
|
--early-stop-patience=10 \
|
|
--datetime-col="date" \
|
|
--use-lora \
|
|
--lora-rank=1 \
|
|
--lora-target-modules="all" \
|
|
--use-dora \
|
|
--cos-initial-decay-value=1e-4 \
|
|
--cos-decay-steps=40000 \
|
|
--cos-final-decay-value=1e-5 \
|
|
--ema-decay=0.9999
|
|
|
|
# To see all available options and their descriptions, use the --help flag
|
|
# python3 finetune.py --help
|