From ab284dc2fcedf9fa286a783cb97d9be14f1413ab Mon Sep 17 00:00:00 2001 From: Chang Xu Date: Sat, 19 Feb 2022 19:06:19 +0800 Subject: [PATCH] fix_demo (#998) --- paddleslim/nas/one_shot/super_mnasnet.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/paddleslim/nas/one_shot/super_mnasnet.py b/paddleslim/nas/one_shot/super_mnasnet.py index 852b4038..169d1050 100644 --- a/paddleslim/nas/one_shot/super_mnasnet.py +++ b/paddleslim/nas/one_shot/super_mnasnet.py @@ -2,7 +2,7 @@ import paddle from paddle import fluid from paddle.fluid.layer_helper import LayerHelper import numpy as np -from one_shot_nas import OneShotSuperNet +from .one_shot_nas import OneShotSuperNet __all__ = ['SuperMnasnet'] @@ -209,14 +209,14 @@ class SuperMnasnet(OneShotSuperNet): def init_tokens(self): return [ - 3, 3, 6, 6, 6, 6, 3, 3, 3, 6, 6, 6, 3, 3, 3, 3, 6, 6, 3, 3, 3, 6, - 6, 6, 3, 3, 3, 6, 6, 6, 3, 6, 6, 6, 6, 6 + 3, 3, 6, 6, 6, 6, 3, 3, 3, 6, 6, 6, 3, 3, 3, 3, 6, 6, 3, 3, 3, 6, 6, + 6, 3, 3, 3, 6, 6, 6, 3, 6, 6, 6, 6, 6 ] def range_table(self): max_v = [ - 6, 6, 10, 10, 10, 10, 6, 6, 6, 10, 10, 10, 6, 6, 6, 6, 10, 10, 6, - 6, 6, 10, 10, 10, 6, 6, 6, 10, 10, 10, 6, 10, 10, 10, 10, 10 + 6, 6, 10, 10, 10, 10, 6, 6, 6, 10, 10, 10, 6, 6, 6, 6, 10, 10, 6, 6, + 6, 10, 10, 10, 6, 6, 6, 10, 10, 10, 6, 10, 10, 10, 10, 10 ] return (len(max_v) * [0], max_v) -- GitLab