Skip to content

mattgorb/sparse-binary-transformers

Repository files navigation

Sparse Binary Transformers

Published at KDD 2023

https://dl.acm.org/doi/pdf/10.1145/3580305.3599508

Experiment scripts in "run_*.sh" files

For example, to train the SMD anomaly detection model run the following:

python3 -u main_ts.py --config=configs/sparse_ts_smd_anomaly.yaml --gpu=6 --lin_prune_rate=0.25 --attention_prune_rate=0.25

Original paper uses two files, main_ts.py, which has single-step forecasting and anomaly detection, and main_ts_classification.py, which has classification.

nohup python -u main_ts.py --config=configs/sparse_ts_ettm1_forecast.yaml --gpu=0 --lin_prune_rate=0.5 --attention_prune_rate=0.5 > ~/results_forecasting/3.txt 2>&1 &

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published