2.0.0 initial
This commit is contained in:
@@ -1,31 +0,0 @@
|
||||
#!/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
|
||||
Reference in New Issue
Block a user