This commit is contained in:
Varuna Jayasiri
2025-07-31 08:48:07 +05:30
parent 00f8714843
commit c4d2e8cd22
101 changed files with 11682 additions and 7441 deletions

View File

@ -82,7 +82,7 @@
<span class="lineno">16</span>
<span class="lineno">17</span><span class="kn">from</span> <span class="nn">labml</span> <span class="kn">import</span> <span class="n">experiment</span><span class="p">,</span> <span class="n">tracker</span>
<span class="lineno">18</span><span class="kn">from</span> <span class="nn">labml.configs</span> <span class="kn">import</span> <span class="n">option</span><span class="p">,</span> <span class="n">calculate</span>
<span class="lineno">19</span><span class="kn">from</span> <span class="nn">labml_helpers.datasets.text</span> <span class="kn">import</span> <span class="n">SequentialUnBatchedDataset</span>
<span class="lineno">19</span><span class="kn">from</span> <span class="nn">labml_nn.helpers.datasets</span> <span class="kn">import</span> <span class="n">SequentialUnBatchedDataset</span>
<span class="lineno">20</span><span class="kn">from</span> <span class="nn">labml_nn.transformers.alibi</span> <span class="kn">import</span> <span class="n">AlibiMultiHeadAttention</span>
<span class="lineno">21</span><span class="kn">from</span> <span class="nn">labml_nn.experiments.nlp_autoregression</span> <span class="kn">import</span> <span class="n">transpose_batch</span>
<span class="lineno">22</span><span class="kn">from</span> <span class="nn">labml_nn.transformers</span> <span class="kn">import</span> <span class="n">TransformerConfigs</span>