deploy: f695c67

ahrefs · Jul 3, 2024 · ab82675 · ab82675
1 parent 5e2851c
commit ab82675
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dev/neural_nets_lib/Ocannl/Train/index.html b/dev/neural_nets_lib/Ocannl/Train/index.html
@@ -48,6 +48,7 @@
   <span><span>(<span class="keyword">module</span> <a href="../../../arrayjit/Arrayjit/Backends/module-type-Backend/index.html">Backend_type</a> <span class="keyword">with</span> <span class="keyword">type</span> <a href="../../../arrayjit/Arrayjit/Backends/module-type-Backend/index.html#type-context">context</a> = <span class="type-var">'context</span>)</span> <span class="arrow">&#45;&gt;</span></span>
   <span><span class="label">grad_updates</span>:<span><span><span class="type-var">'context0</span> <a href="../../../arrayjit/Arrayjit/Backend_types/index.html#type-routine">Arrayjit.Backend_types.routine</a></span> <span class="xref-unresolved">Base</span>.array</span> <span class="arrow">&#45;&gt;</span></span>
   <span><span class="label">sgd_update</span>:<span><span class="type-var">'context0</span> <a href="../../../arrayjit/Arrayjit/Backend_types/index.html#type-routine">Arrayjit.Backend_types.routine</a></span> <span class="arrow">&#45;&gt;</span></span>
+  <span><span class="label">copy_to_merge</span>:bool <span class="arrow">&#45;&gt;</span></span>
   <span><span class="label">post_sync</span>:<span>(<span><span class="label">num_synced_devices</span>:<span class="xref-unresolved">Base</span>.int <span class="arrow">&#45;&gt;</span></span> <span class="xref-unresolved">Base</span>.unit)</span> <span class="arrow">&#45;&gt;</span></span>
   <span><a href="#type-updaten">updaten</a> <span class="arrow">&#45;&gt;</span></span>
   <span><span class="xref-unresolved">Base</span>.unit <span class="arrow">&#45;&gt;</span></span>
@@ -61,6 +62,7 @@
   <span><span class="label">init_lr</span>:<span class="xref-unresolved">Base</span>.float <span class="arrow">&#45;&gt;</span></span>
   <span><span class="optlabel">?lr_schedule</span>:
     <span>(<span><span class="label">batch_n</span>:<a href="../../../arrayjit/Arrayjit/Indexing/index.html#type-static_symbol">Idx.static_symbol</a> <span class="arrow">&#45;&gt;</span></span> <span><span class="label">step_n</span>:<a href="../../../arrayjit/Arrayjit/Indexing/index.html#type-static_symbol">Idx.static_symbol</a> <span class="arrow">&#45;&gt;</span></span> <a href="../Tensor/index.html#type-t">Tensor.t</a>)</span> <span class="arrow">&#45;&gt;</span></span>
+  <span><span class="optlabel">?copy_to_merge</span>:bool <span class="arrow">&#45;&gt;</span></span>
   <span><span class="optlabel">?max_num_devices</span>:<span class="xref-unresolved">Base</span>.Int.t <span class="arrow">&#45;&gt;</span></span>
   <span><span class="label">data_len</span>:<span class="xref-unresolved">Base__Int</span>.t <span class="arrow">&#45;&gt;</span></span>
   <span><span class="label">epochs</span>:<span class="xref-unresolved">Base__Int</span>.t <span class="arrow">&#45;&gt;</span></span>