mirror of
https://github.com/marian-nmt/marian.git
synced 2024-08-15 01:00:23 +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,) ``` |
||
|---|---|---|
| .. | ||
| bert | ||
| checkpoints | ||
| ci | ||
| contrib | ||
| embeddings | ||
| laser | ||
| onnx | ||
| server | ||
| shortlist | ||