This commit is contained in:
Varuna Jayasiri
2021-02-19 09:31:07 +05:30
parent ccb9ee2e4c
commit 6a5df4fc11
2 changed files with 2 additions and 2 deletions

View File

@ -1242,7 +1242,7 @@ Memories that were compressed is needed for the reconstruction loss computation.
<span class="lineno">335</span> <span class="s1">&#39;epochs&#39;</span><span class="p">:</span> <span class="mi">128</span><span class="p">,</span> <span class="lineno">335</span> <span class="s1">&#39;epochs&#39;</span><span class="p">:</span> <span class="mi">128</span><span class="p">,</span>
<span class="lineno">336</span> <span class="s1">&#39;batch_size&#39;</span><span class="p">:</span> <span class="mi">32</span><span class="p">,</span> <span class="lineno">336</span> <span class="s1">&#39;batch_size&#39;</span><span class="p">:</span> <span class="mi">32</span><span class="p">,</span>
<span class="lineno">337</span> <span class="s1">&#39;inner_iterations&#39;</span><span class="p">:</span> <span class="mi">25</span><span class="p">,</span> <span class="lineno">337</span> <span class="s1">&#39;inner_iterations&#39;</span><span class="p">:</span> <span class="mi">25</span><span class="p">,</span>
<span class="lineno">338</span> <span class="s1">&#39;compression_ratio&#39;</span><span class="p">:</span> <span class="mi">2</span><span class="p">,</span> <span class="lineno">338</span> <span class="s1">&#39;compression_rate&#39;</span><span class="p">:</span> <span class="mi">2</span><span class="p">,</span>
<span class="lineno">339</span> <span class="p">})</span></pre></div> <span class="lineno">339</span> <span class="p">})</span></pre></div>
</div> </div>
</div> </div>

View File

@ -335,7 +335,7 @@ def main():
'epochs': 128, 'epochs': 128,
'batch_size': 32, 'batch_size': 32,
'inner_iterations': 25, 'inner_iterations': 25,
'compression_ratio': 2, 'compression_rate': 2,
}) })
# Set models for saving and loading # Set models for saving and loading