diff --git a/.DS_Store b/.DS_Store new file mode 100644 index 000000000..b7fd12f31 Binary files /dev/null and b/.DS_Store differ diff --git a/qlib/contrib/model/pytorch_gats_ts.py b/qlib/contrib/model/pytorch_gats_ts.py index f5d8fefaa..401719275 100644 --- a/qlib/contrib/model/pytorch_gats_ts.py +++ b/qlib/contrib/model/pytorch_gats_ts.py @@ -73,7 +73,7 @@ class GATs(Model): base_model="GRU", model_path=None, optimizer="adam", - GPU="0", + GPU=0, n_jobs=10, seed=None, **kwargs