This commit is contained in:
Varuna Jayasiri
2022-06-06 10:04:59 +05:30
parent 680b8e5734
commit feaa95574e
2 changed files with 3 additions and 3 deletions

View File

@ -83,7 +83,7 @@
<span class="lineno">19</span><span class="kn">from</span> <span class="nn">labml.logger</span> <span class="kn">import</span> <span class="n">Text</span>
<span class="lineno">20</span><span class="kn">from</span> <span class="nn">labml_helpers.datasets.text</span> <span class="kn">import</span> <span class="n">TextDataset</span><span class="p">,</span> <span class="n">SequentialDataLoader</span><span class="p">,</span> <span class="n">SequentialUnBatchedDataset</span><span class="p">,</span> <span class="n">TextFileDataset</span>
<span class="lineno">21</span><span class="kn">from</span> <span class="nn">labml_helpers.device</span> <span class="kn">import</span> <span class="n">DeviceConfigs</span>
<span class="lineno">22</span><span class="kn">from</span> <span class="nn">labml_helpers.metrics.accuracy</span> <span class="kn">import</span> <span class="n">Accuracy</span>
<span class="lineno">22</span><span class="kn">from</span> <span class="nn">labml_helpers.metrics.accuracy</span> <span class="kn">import</span> <span class="n">Accuracy</span><span class="p">,</span> <span class="n">AccuracyMovingAvg</span>
<span class="lineno">23</span><span class="kn">from</span> <span class="nn">labml_helpers.module</span> <span class="kn">import</span> <span class="n">Module</span>
<span class="lineno">24</span><span class="kn">from</span> <span class="nn">labml_helpers.train_valid</span> <span class="kn">import</span> <span class="n">TrainValidConfigs</span><span class="p">,</span> <span class="n">hook_model_outputs</span><span class="p">,</span> <span class="n">BatchIndex</span>
<span class="lineno">25</span><span class="kn">from</span> <span class="nn">labml_nn.optimizers.configs</span> <span class="kn">import</span> <span class="n">OptimizerConfigs</span></pre></div>
@ -293,7 +293,7 @@
</div>
<div class='code'>
<div class="highlight"><pre><span class="lineno">80</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="n">Accuracy</span><span class="p">()</span></pre></div>
<div class="highlight"><pre><span class="lineno">80</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="n">AccuracyMovingAvg</span><span class="p">()</span></pre></div>
</div>
</div>
<div class='section' id='section-18'>

View File

@ -21,7 +21,7 @@ setuptools.setup(
'test',
'test.*')),
install_requires=['labml>=0.4.151',
'labml-helpers>=0.4.86',
'labml-helpers>=0.4.87',
'torch',
'torchtext',
'torchvision',