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">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">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>
</div>
</div>

View File

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