deploy: 263b2cb

dice-group · Sep 21, 2024 · 6036b40 · 6036b40
1 parent 3a48bc8
commit 6036b40
Show file tree

Hide file tree

Showing 7 changed files with 24 additions and 267 deletions.
diff --git a/_modules/dicee/trainer/torch_trainer_ddp.html b/_modules/dicee/trainer/torch_trainer_ddp.html
diff --git a/_sources/autoapi/dicee/trainer/torch_trainer_ddp/index.rst.txt b/_sources/autoapi/dicee/trainer/torch_trainer_ddp/index.rst.txt
@@ -11,7 +11,6 @@ Classes
 
    dicee.trainer.torch_trainer_ddp.TorchDDPTrainer
    dicee.trainer.torch_trainer_ddp.NodeTrainer
-   dicee.trainer.torch_trainer_ddp.DDPTrainer
 
 
 Functions
@@ -62,7 +61,7 @@ Module Contents
 
 
 
-.. py:class:: NodeTrainer(trainer, model: torch.nn.Module, train_dataset_loader: torch.utils.data.DataLoader, optimizer: torch.optim.Optimizer, callbacks, num_epochs: int)
+.. py:class:: NodeTrainer(trainer, model: torch.nn.Module, train_dataset_loader: torch.utils.data.DataLoader, callbacks, num_epochs: int)
 
    .. py:attribute:: trainer
 
@@ -73,7 +72,7 @@ Module Contents
    .. py:attribute:: global_rank
 
 
-   .. py:attribute:: model
+   .. py:attribute:: optimizer
 
 
    .. py:attribute:: train_dataset_loader
@@ -82,10 +81,10 @@ Module Contents
    .. py:attribute:: loss_func
 
 
-   .. py:attribute:: optimizer
+   .. py:attribute:: callbacks
 
 
-   .. py:attribute:: callbacks
+   .. py:attribute:: model
 
 
    .. py:attribute:: num_epochs
@@ -105,37 +104,3 @@ Module Contents
 
 
 
-.. py:class:: DDPTrainer(model: torch.nn.Module, train_dataset_loader: torch.utils.data.DataLoader, optimizer: torch.optim.Optimizer, gpu_id: int, callbacks, num_epochs)
-
-   .. py:attribute:: gpu_id
-
-
-   .. py:attribute:: model
-
-
-   .. py:attribute:: train_dataset_loader
-
-
-   .. py:attribute:: loss_func
-
-
-   .. py:attribute:: optimizer
-
-
-   .. py:attribute:: callbacks
-
-
-   .. py:attribute:: num_epochs
-
-
-   .. py:attribute:: loss_history
-      :value: []
-
-
-
-   .. py:method:: extract_input_outputs(z: list)
-
-
-   .. py:method:: train()
-
-
diff --git a/autoapi/dicee/trainer/torch_trainer_ddp/index.html b/autoapi/dicee/trainer/torch_trainer_ddp/index.html
@@ -135,9 +135,6 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">
 <tr class="row-even"><td><p><a class="reference internal" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer" title="dicee.trainer.torch_trainer_ddp.NodeTrainer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">NodeTrainer</span></code></a></p></td>
 <td><p></p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer" title="dicee.trainer.torch_trainer_ddp.DDPTrainer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">DDPTrainer</span></code></a></p></td>
-<td><p></p></td>
-</tr>
 </tbody>
 </table>
 </section>
@@ -202,7 +199,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 
 <dl class="py class">
 <dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.NodeTrainer">
-<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">dicee.trainer.torch_trainer_ddp.</span></span><span class="sig-name descname"><span class="pre">NodeTrainer</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">trainer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">model</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">torch.nn.Module</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">train_dataset_loader</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">torch.utils.data.DataLoader</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">optimizer</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">torch.optim.Optimizer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">callbacks</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">num_epochs</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">int</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../../../../_modules/dicee/trainer/torch_trainer_ddp.html#NodeTrainer"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer" title="Link to this definition"></a></dt>
+<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">dicee.trainer.torch_trainer_ddp.</span></span><span class="sig-name descname"><span class="pre">NodeTrainer</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">trainer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">model</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">torch.nn.Module</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">train_dataset_loader</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">torch.utils.data.DataLoader</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">callbacks</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">num_epochs</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">int</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../../../../_modules/dicee/trainer/torch_trainer_ddp.html#NodeTrainer"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer" title="Link to this definition"></a></dt>
 <dd><dl class="py attribute">
 <dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.NodeTrainer.trainer">
 <span class="sig-name descname"><span class="pre">trainer</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer.trainer" title="Link to this definition"></a></dt>
@@ -219,8 +216,8 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dd></dd></dl>
 
 <dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.NodeTrainer.model">
-<span class="sig-name descname"><span class="pre">model</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer.model" title="Link to this definition"></a></dt>
+<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.NodeTrainer.optimizer">
+<span class="sig-name descname"><span class="pre">optimizer</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer.optimizer" title="Link to this definition"></a></dt>
 <dd></dd></dl>
 
 <dl class="py attribute">
@@ -234,13 +231,13 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dd></dd></dl>
 
 <dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.NodeTrainer.optimizer">
-<span class="sig-name descname"><span class="pre">optimizer</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer.optimizer" title="Link to this definition"></a></dt>
+<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.NodeTrainer.callbacks">
+<span class="sig-name descname"><span class="pre">callbacks</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer.callbacks" title="Link to this definition"></a></dt>
 <dd></dd></dl>
 
 <dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.NodeTrainer.callbacks">
-<span class="sig-name descname"><span class="pre">callbacks</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer.callbacks" title="Link to this definition"></a></dt>
+<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.NodeTrainer.model">
+<span class="sig-name descname"><span class="pre">model</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.NodeTrainer.model" title="Link to this definition"></a></dt>
 <dd></dd></dl>
 
 <dl class="py attribute">
@@ -266,61 +263,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 
 </dd></dl>
 
-<dl class="py class">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer">
-<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">dicee.trainer.torch_trainer_ddp.</span></span><span class="sig-name descname"><span class="pre">DDPTrainer</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">model</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">torch.nn.Module</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">train_dataset_loader</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">torch.utils.data.DataLoader</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">optimizer</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">torch.optim.Optimizer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">gpu_id</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">callbacks</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">num_epochs</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../../../../_modules/dicee/trainer/torch_trainer_ddp.html#DDPTrainer"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer" title="Link to this definition"></a></dt>
-<dd><dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.gpu_id">
-<span class="sig-name descname"><span class="pre">gpu_id</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.gpu_id" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-<dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.model">
-<span class="sig-name descname"><span class="pre">model</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.model" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-<dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.train_dataset_loader">
-<span class="sig-name descname"><span class="pre">train_dataset_loader</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.train_dataset_loader" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-<dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.loss_func">
-<span class="sig-name descname"><span class="pre">loss_func</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.loss_func" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-<dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.optimizer">
-<span class="sig-name descname"><span class="pre">optimizer</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.optimizer" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-<dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.callbacks">
-<span class="sig-name descname"><span class="pre">callbacks</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.callbacks" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-<dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.num_epochs">
-<span class="sig-name descname"><span class="pre">num_epochs</span></span><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.num_epochs" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-<dl class="py attribute">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.loss_history">
-<span class="sig-name descname"><span class="pre">loss_history</span></span><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">[]</span></em><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.loss_history" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-<dl class="py method">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.extract_input_outputs">
-<span class="sig-name descname"><span class="pre">extract_input_outputs</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">z</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">list</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../../../../_modules/dicee/trainer/torch_trainer_ddp.html#DDPTrainer.extract_input_outputs"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.extract_input_outputs" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-<dl class="py method">
-<dt class="sig sig-object py" id="dicee.trainer.torch_trainer_ddp.DDPTrainer.train">
-<span class="sig-name descname"><span class="pre">train</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="reference internal" href="../../../../_modules/dicee/trainer/torch_trainer_ddp.html#DDPTrainer.train"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dicee.trainer.torch_trainer_ddp.DDPTrainer.train" title="Link to this definition"></a></dt>
-<dd></dd></dl>
-
-</dd></dl>
-
 </section>
 </section>
 

diff --git a/diceembeddings.pdf b/diceembeddings.pdf