diff --git a/lib/xlayers/super_attention.py b/lib/xlayers/super_attention.py
index d23b08a..cddb3a5 100644
--- a/lib/xlayers/super_attention.py
+++ b/lib/xlayers/super_attention.py
@@ -113,7 +113,7 @@ class SuperAttention(SuperModule):
             .permute(0, 2, 1, 3)
         )
         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)
         feats_v1 = (attn_v1 @ v_v1).permute(0, 2, 1, 3).reshape(B, N, -1)
         if C == head_dim * num_head:
diff --git a/scripts/trade/baseline.sh b/scripts/trade/baseline.sh
index 05a0de8..438c3de 100644
--- a/scripts/trade/baseline.sh
+++ b/scripts/trade/baseline.sh
@@ -16,11 +16,10 @@ fi
 gpu=$1
 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}
 do
-
   python exps/trading/baselines.py --alg ${alg} --gpu ${gpu} --market ${market}
-
 done