From 183b109a03755999c2d52f162ade351f62162474 Mon Sep 17 00:00:00 2001 From: Duo <50307526+iProzd@users.noreply.github.com> Date: Fri, 10 May 2024 21:04:54 +0800 Subject: [PATCH] Update test_unused_params.py --- source/tests/pt/model/test_unused_params.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/source/tests/pt/model/test_unused_params.py b/source/tests/pt/model/test_unused_params.py index 730eb00a0b..b4c69ab69f 100644 --- a/source/tests/pt/model/test_unused_params.py +++ b/source/tests/pt/model/test_unused_params.py @@ -41,15 +41,15 @@ def test_unused(self): # skip the case g2 is not envolved continue model = copy.deepcopy(model_dpa2) - model["descriptor"]["repformer_nlayers"] = 2 + model["descriptor"]["repformer"]["nlayers"] = 2 # model["descriptor"]["combine_grrg"] = cmbg2 - model["descriptor"]["repformer_update_g1_has_conv"] = conv - model["descriptor"]["repformer_update_g1_has_drrd"] = drrd - model["descriptor"]["repformer_update_g1_has_grrg"] = grrg - model["descriptor"]["repformer_update_g1_has_attn"] = attn1 - model["descriptor"]["repformer_update_g2_has_g1g1"] = g1g1 - model["descriptor"]["repformer_update_g2_has_attn"] = attn2 - model["descriptor"]["repformer_update_h2"] = h2 + model["descriptor"]["repformer"]["update_g1_has_conv"] = conv + model["descriptor"]["repformer"]["update_g1_has_drrd"] = drrd + model["descriptor"]["repformer"]["update_g1_has_grrg"] = grrg + model["descriptor"]["repformer"]["update_g1_has_attn"] = attn1 + model["descriptor"]["repformer"]["update_g2_has_g1g1"] = g1g1 + model["descriptor"]["repformer"]["update_g2_has_attn"] = attn2 + model["descriptor"]["repformer"]["update_h2"] = h2 model["fitting_net"]["neuron"] = [12, 12, 12] self._test_unused(model)