Test PP-FormulaNet-L head. Args: encoder_feat_ppformulanet_l: encoder feature from PP-FormulaNet-L Head.
(encoder_feat_ppformulanet_l)
| 174 | |
| 175 | |
| 176 | def test_ppformulanet_l_head(encoder_feat_ppformulanet_l): |
| 177 | """ |
| 178 | Test PP-FormulaNet-L head. |
| 179 | |
| 180 | Args: |
| 181 | encoder_feat_ppformulanet_l: encoder feature from PP-FormulaNet-L Head. |
| 182 | """ |
| 183 | head = PPFormulaNet_Head( |
| 184 | max_new_tokens=6, |
| 185 | decoder_start_token_id=0, |
| 186 | decoder_ffn_dim=2048, |
| 187 | decoder_hidden_size=512, |
| 188 | decoder_layers=8, |
| 189 | temperature=0.2, |
| 190 | do_sample=False, |
| 191 | top_p=0.95, |
| 192 | encoder_hidden_size=1024, |
| 193 | is_export=False, |
| 194 | length_aware=False, |
| 195 | use_parallel=False, |
| 196 | parallel_step=0, |
| 197 | ) |
| 198 | |
| 199 | head.eval() |
| 200 | with paddle.no_grad(): |
| 201 | result = head(encoder_feat_ppformulanet_l) |
| 202 | assert result.shape == [1, 7] |
nothing calls this directly
no test coverage detected
searching dependent graphs…