chamgj
/
netrans
forked from nudt_dsp/netrans
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
netrans/docs/html/_modules/config.html

166 lines
7.9 KiB
HTML
Raw Blame History

<!DOCTYPE html>
<html lang="zh" data-content_root="../">
<head>
<meta charset="utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>config &#8212; netrans 0.1 文档</title>
<link rel="stylesheet" type="text/css" href="../_static/pygments.css?v=5ecbeea2" />
<link rel="stylesheet" type="text/css" href="../_static/basic.css?v=b08954a9" />
<link rel="stylesheet" type="text/css" href="../_static/alabaster.css?v=27fed22d" />
<script src="../_static/documentation_options.js?v=52efc512"></script>
<script src="../_static/doctools.js?v=9bcbadda"></script>
<script src="../_static/sphinx_highlight.js?v=dc90522c"></script>
<link rel="index" title="索引" href="../genindex.html" />
<link rel="search" title="搜索" href="../search.html" />
<link rel="stylesheet" href="../_static/custom.css" type="text/css" />
</head><body>
<div class="document">
<div class="documentwrapper">
<div class="bodywrapper">
<div class="body" role="main">
<h1>config 源代码</h1><div class="highlight"><pre>
<span></span>
<span class="kn">import</span><span class="w"> </span><span class="nn">os</span>
<span class="kn">import</span><span class="w"> </span><span class="nn">sys</span>
<span class="kn">from</span><span class="w"> </span><span class="nn">utils</span><span class="w"> </span><span class="kn">import</span> <span class="n">check_path</span><span class="p">,</span> <span class="n">AttributeCopier</span><span class="p">,</span> <span class="n">create_cls</span>
<span class="kn">import</span><span class="w"> </span><span class="nn">subprocess</span>
<div class="viewcode-block" id="Config">
<a class="viewcode-back" href="../config.html#config.Config">[文档]</a>
<span class="k">class</span><span class="w"> </span><span class="nc">Config</span><span class="p">(</span><span class="n">AttributeCopier</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;从实例化的 Netrans 中解析模型参数,并基于pnnacc 生成配置文件模板</span>
<span class="sd"> Args:</span>
<span class="sd"> Netrans (class): 实例化的Netrans类,包含 模型信息 和 Netrans 信息</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">def</span><span class="w"> </span><span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">source_obj</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;从实例化的 Netrans 中解析模型参数</span>
<span class="sd"> Args:</span>
<span class="sd"> source_obj (class): 实例化的Netrans类,包含 模型信息 和 Netrans 信息</span>
<span class="sd"> </span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="n">source_obj</span><span class="p">)</span>
<span class="nd">@check_path</span>
<span class="k">def</span><span class="w"> </span><span class="nf">inputmeta_gen</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;生成配置文件模板</span>
<span class="sd"> Return:</span>
<span class="sd"> None</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">netrans_path</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">netrans</span>
<span class="n">network_name</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">model_name</span>
<span class="c1"># 进入网络名称指定的目录</span>
<span class="c1"># os.chdir(network_name)</span>
<span class="c1"># check_env(network_name)</span>
<span class="c1"># 执行 pegasus 命令</span>
<span class="n">cmd</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">&quot;</span><span class="si">{</span><span class="n">netrans_path</span><span class="si">}</span><span class="s2"> generate inputmeta --model </span><span class="si">{</span><span class="n">network_name</span><span class="si">}</span><span class="s2">.json --separated-database&quot;</span>
<span class="k">try</span> <span class="p">:</span>
<span class="n">result</span> <span class="o">=</span> <span class="n">subprocess</span><span class="o">.</span><span class="n">run</span><span class="p">(</span><span class="n">cmd</span><span class="p">,</span> <span class="n">shell</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">capture_output</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">text</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
<span class="k">except</span> <span class="p">:</span>
<span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">(</span><span class="s1">&#39;config failed&#39;</span><span class="p">)</span></div>
<span class="c1"># os.chdir(&quot;..&quot;)</span>
<span class="c1"># def main():</span>
<span class="c1"># # 检查命令行参数数量是否正确</span>
<span class="c1"># if len(sys.argv) != 2:</span>
<span class="c1"># print(&quot;Enter a network name!&quot;)</span>
<span class="c1"># sys.exit(2)</span>
<span class="c1"># # 检查提供的目录是否存在</span>
<span class="c1"># network_name = sys.argv[1]</span>
<span class="c1"># # 构建 netrans 可执行文件的路径</span>
<span class="c1"># netrans_path =os.getenv(&#39;NETRANS_PATH&#39;)</span>
<span class="c1"># cla = create_cls(netrans_path, network_name)</span>
<span class="c1"># func = InputmetaGen(cla)</span>
<span class="c1"># func.inputmeta_gen()</span>
<span class="c1"># if __name__ == &#39;__main__&#39;:</span>
<span class="c1"># main()</span>
</pre></div>
</div>
</div>
</div>
<div class="sphinxsidebar" role="navigation" aria-label="Main">
<div class="sphinxsidebarwrapper">
<h1 class="logo"><a href="../index.html">netrans</a></h1>
<search id="searchbox" style="display: none" role="search">
<div class="searchformwrapper">
<form class="search" action="../search.html" method="get">
<input type="text" name="q" aria-labelledby="searchlabel" autocomplete="off" autocorrect="off" autocapitalize="off" spellcheck="false" placeholder="Search"/>
<input type="submit" value="提交" />
</form>
</div>
</search>
<script>document.getElementById('searchbox').style.display = "block"</script><h3>导航</h3>
<p class="caption" role="heading"><span class="caption-text">Contents:</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../quick_start_guide.html">快速入门</a></li>
<li class="toctree-l1"><a class="reference internal" href="../netrans_cli.html">netrans_cli 使用</a></li>
<li class="toctree-l1"><a class="reference internal" href="../netrans_py.html">netrans_py 使用</a></li>
<li class="toctree-l1"><a class="reference internal" href="../appendix.html">附录</a></li>
</ul>
<div class="relations">
<h3>Related Topics</h3>
<ul>
<li><a href="../index.html">Documentation overview</a><ul>
<li><a href="index.html">模块代码</a><ul>
</ul></li>
</ul></li>
</ul>
</div>
</div>
</div>
<div class="clearer"></div>
</div>
<div class="footer">
&#169;2025, ccyh.
|
Powered by <a href="https://www.sphinx-doc.org/">Sphinx 8.2.3</a>
&amp; <a href="https://alabaster.readthedocs.io">Alabaster 1.0.0</a>
</div>
</body>
</html>
Reference in New Issue View Git Blame Copy Permalink