</span>
<span class="n">params</span> <span class="o">=</span> <span class="n">test_base</span><span class="o">.</span><span class="n">params</span>

<span class="n">hidden_dim</span> <span class="o">=</span> <span class="n">params</span><span class="o">.</span><span class="n">bert_config</span><span class="o">.</span><span class="n">hidden_size</span>

<span class="n">train_dataset</span> <span class="o">=</span> <span class="n">train_eval_input_fn</span><span class="p">(</span><span class="n">params</span