mirror of
https://github.com/marian-nmt/marian.git
synced 2024-10-05 19:17:10 +03:00
a90950ea25
Add model shapes flag to model_info.py script through `--matrix-shapes` flag This will print something like: ``` ... encoder_l6_ffn_W1 (1024, 4096) encoder_l6_ffn_W2 (4096, 1024) encoder_l6_ffn_b1 (1, 4096) encoder_l6_ffn_b2 (1, 1024) encoder_l6_ffn_ffn_ln_bias (1, 1024) encoder_l6_ffn_ffn_ln_scale (1, 1024) encoder_l6_self_Wk (1024, 1024) encoder_l6_self_Wo (1024, 1024) encoder_l6_self_Wo_ln_bias (1, 1024) encoder_l6_self_Wo_ln_scale (1, 1024) encoder_l6_self_Wq (1024, 1024) encoder_l6_self_Wv (1024, 1024) encoder_l6_self_bk (1, 1024) encoder_l6_self_bo (1, 1024) encoder_l6_self_bq (1, 1024) encoder_l6_self_bv (1, 1024) special:model.yml (1264,) ``` |
||
---|---|---|
.. | ||
fix_hard.py | ||
inject_ctt.py | ||
inject_model_params.py | ||
model_info.py |