()
| 32 | |
| 33 | |
| 34 | def init_decoder_model(): |
| 35 | return TransformerDecoder( |
| 36 | vocab_size=10, |
| 37 | encoder_output_size=40, |
| 38 | attention_heads=4, |
| 39 | linear_units=40, |
| 40 | num_blocks=2, |
| 41 | input_layer="embed", |
| 42 | ) |
| 43 | |
| 44 | |
| 45 | # =========================================Houlsby================================================ |
no test coverage detected
searching dependent graphs…