|
12 | 12 | <meta name="twitter:site" content="@labmlai"/>
|
13 | 13 | <meta name="twitter:creator" content="@labmlai"/>
|
14 | 14 |
|
15 |
| - <meta property="og:url" content="https://nn.labml.ai/RWKV/experiment.html"/> |
| 15 | + <meta property="og:url" content="https://nn.labml.ai/rwkv/experiment.html"/> |
16 | 16 | <meta property="og:title" content="experiment.py"/>
|
17 | 17 | <meta property="og:image" content="https://avatars1.githubusercontent.com/u/64068543?s=400&v=4"/>
|
18 | 18 | <meta property="og:site_name" content="experiment.py"/>
|
|
23 | 23 | <title>experiment.py</title>
|
24 | 24 | <link rel="shortcut icon" href="/icon.png"/>
|
25 | 25 | <link rel="stylesheet" href="../pylit.css?v=1">
|
26 |
| - <link rel="canonical" href="https://nn.labml.ai/RWKV/experiment.html"/> |
| 26 | + <link rel="canonical" href="https://nn.labml.ai/rwkv/experiment.html"/> |
27 | 27 | <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/katex@0.13.18/dist/katex.min.css" integrity="sha384-zTROYFVGOfTw7JV7KUu8udsvW2fx4lWOsCEDqhBreBwlHI4ioVRtmIvEThzJHGET" crossorigin="anonymous">
|
28 | 28 |
|
29 | 29 | <!-- Global site tag (gtag.js) - Google Analytics -->
|
|
47 | 47 | <div class='docs'>
|
48 | 48 | <p>
|
49 | 49 | <a class="parent" href="/">home</a>
|
50 |
| - <a class="parent" href="index.html">RWKV</a> |
| 50 | + <a class="parent" href="index.html">rwkv</a> |
51 | 51 | </p>
|
52 | 52 | <p>
|
53 | 53 | <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations" target="_blank">
|
|
60 | 60 | style="max-width:100%;"/></a>
|
61 | 61 | </p>
|
62 | 62 | <p>
|
63 |
| - <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations/tree/master/labml_nn/RWKV/experiment.py" target="_blank"> |
| 63 | + <a href="https://github.com/labmlai/annotated_deep_learning_paper_implementations/tree/master/labml_nn/rwkv/experiment.py" target="_blank"> |
64 | 64 | View code on Github</a>
|
65 | 65 | </p>
|
66 | 66 | </div>
|
|
78 | 78 | <span class="lineno">3</span>
|
79 | 79 | <span class="lineno">4</span><span class="kn">import</span> <span class="nn">torch</span>
|
80 | 80 | <span class="lineno">5</span><span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
|
81 |
| -<span class="lineno">6</span><span class="kn">from</span> <span class="nn">labml_nn.RWKV.configs</span> <span class="kn">import</span> <span class="n">RWKVConfigs</span> |
| 81 | +<span class="lineno">6</span><span class="kn">from</span> <span class="nn">labml_nn.rwkv.configs</span> <span class="kn">import</span> <span class="n">RWKVConfigs</span> |
82 | 82 | <span class="lineno">7</span>
|
83 |
| -<span class="lineno">8</span><span class="kn">from</span> <span class="nn">labml_nn.RWKV</span> <span class="kn">import</span> <span class="n">RWKV</span> |
84 |
| -<span class="lineno">9</span><span class="kn">from</span> <span class="nn">labml_nn.RWKV</span> <span class="kn">import</span> <span class="n">TimeMixing</span> |
| 83 | +<span class="lineno">8</span><span class="kn">from</span> <span class="nn">labml_nn.rwkv</span> <span class="kn">import</span> <span class="n">RWKV</span> |
| 84 | +<span class="lineno">9</span><span class="kn">from</span> <span class="nn">labml_nn.rwkv</span> <span class="kn">import</span> <span class="n">TimeMixing</span> |
85 | 85 | <span class="lineno">10</span><span class="kn">from</span> <span class="nn">labml</span> <span class="kn">import</span> <span class="n">experiment</span>
|
86 | 86 | <span class="lineno">11</span><span class="kn">from</span> <span class="nn">labml.configs</span> <span class="kn">import</span> <span class="n">option</span>
|
87 | 87 | <span class="lineno">12</span><span class="kn">from</span> <span class="nn">labml_nn.experiments.nlp_autoregression</span> <span class="kn">import</span> <span class="n">NLPAutoRegressionConfigs</span></pre></div>
|
|
0 commit comments