zchen/b2txt25
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

小参数

Browse Source
This commit is contained in:
Zchen
2025-10-16 09:22:25 +08:00
parent 25561a7615
commit df4a914bbd
2 changed files with 206 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
model_training_nnn_tpu/rnn_args.yaml
View File

@@ -1,9 +1,9 @@
model:
n_input_features: 512 # number of input features in the neural data. (2 features per electrode, 256 electrodes)
n_units: 768 # number of units per GRU layer
n_units: 256 # number of units per GRU layer (大幅减少从768→256, 减少70%参数量)
rnn_dropout: 0.4 # dropout rate for the GRU layers
rnn_trainable: true # whether the GRU layers are trainable
n_layers: 5 # number of GRU layers
n_layers: 3 # number of GRU layers (从5层减少到3层)
patch_size: 14 # size of the input patches (14 time steps)
patch_stride: 4 # stride for the input patches (4 time steps)