o
    ߥi                     @   s   d dl mZ dd ZdS )   )
master_netc                  C   s    d} d}d}t j|| |d}|S )Na  SuperConvK3BNRELU(3,32,2,1)SuperResK1K5K1(32,80,2,32,1)SuperResK1K7K1(80,432,2,128,5)SuperResK1K7K1(432,640,2,192,3)SuperResK1K7K1(640,1008,1,160,5)SuperResK1K7K1(1008,976,1,160,4)SuperResK1K5K1(976,2304,2,384,5)SuperResK1K5K1(2304,2496,1,384,5)SuperConvK1BNRELU(2496,3072,1,1)Fi  )num_classesplainnet_structuse_se)r   PlainNet)model_plainnet_struse_SEr   model r
   h/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/cv/tinynas_classfication/model_zoo.py
get_zennet   s   r   N) r   r   r
   r
   r
   r   <module>   s   