mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
Deployed df77f50 with MkDocs version: 1.0.4
This commit is contained in:
@@ -678,7 +678,8 @@ cluster.per_experiment_nb_gpus = 8
|
||||
cluster.add_slurm_cmd(cmd='ntasks-per-node', value=8, comment='1 task per gpu')
|
||||
</code></pre>
|
||||
|
||||
<p>Finally, make sure to add a distributed sampler to your dataset. </p>
|
||||
<p>Finally, make sure to add a distributed sampler to your dataset. The distributed sampler copies a
|
||||
portion of your dataset onto each GPU. (World_size = gpus_per_node * nb_nodes). </p>
|
||||
<pre><code class="python"># ie: this:
|
||||
dataset = myDataset()
|
||||
dataloader = Dataloader(dataset)
|
||||
|
||||
File diff suppressed because one or more lines are too long
Binary file not shown.
Reference in New Issue
Block a user