iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update baseline.sh

Browse Source
This commit is contained in:
D-X-Y
2021-04-02 00:40:26 -07:00
parent 0e35d8b156
commit 1dd665ae06
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/xlayers/super_attention.py
View File

@@ -113,7 +113,7 @@ class SuperAttention(SuperModule):
.permute(0, 2, 1, 3) .permute(0, 2, 1, 3)
) )
attn_v1 = (q_v1 @ k_v1.transpose(-2, -1)) * math.sqrt(head_dim) attn_v1 = (q_v1 @ k_v1.transpose(-2, -1)) * math.sqrt(head_dim)
attn_v1 = attn_v1.softmax(dim=-1) attn_v1 = attn_v1.softmax(dim=-1) # B * #head * N * N
attn_v1 = self.attn_drop(attn_v1) attn_v1 = self.attn_drop(attn_v1)
feats_v1 = (attn_v1 @ v_v1).permute(0, 2, 1, 3).reshape(B, N, -1) feats_v1 = (attn_v1 @ v_v1).permute(0, 2, 1, 3).reshape(B, N, -1)
if C == head_dim * num_head: if C == head_dim * num_head:

5
scripts/trade/baseline.sh
View File

@@ -16,11 +16,10 @@ fi
gpu=$1 gpu=$1
market=$2 market=$2
algorithms="NAIVE-V1 NAIVE-V2 MLP GRU LSTM ALSTM XGBoost LightGBM SFM TabNet DoubleE" # algorithms="NAIVE-V1 NAIVE-V2 MLP GRU LSTM ALSTM XGBoost LightGBM SFM TabNet DoubleE"
algorithms="MLP GRU LSTM ALSTM XGBoost LightGBM SFM TabNet DoubleE"
for alg in ${algorithms} for alg in ${algorithms}
do do
python exps/trading/baselines.py --alg ${alg} --gpu ${gpu} --market ${market} python exps/trading/baselines.py --alg ${alg} --gpu ${gpu} --market ${market}
done done