class_docstr: Specific tests on SimCLR going further than just doing a forward pass
class_docstr_tok: ["Specific","tests","on","SimCLR","going","further","than","just","doing","a","forward","pass"]
code: def test_gather_embeddings_word_size_1(self) -> None:
      with with_temp_files(count=1) as sync_file:
          WORLD_SIZE = 1
          BATCH_SIZE = 2
          mp.spawn(
              self.worker_fn,
              args=(WORLD_SIZE, BATCH_SIZE, sync_file),
              nprocs=WORLD_SIZE,
          )
  
file_path: vissl/tests/test_losses_gpu.py
imports: import unittest
  import torch
  import torch.multiprocessing
  from vissl.losses.barlow_twins_loss import BarlowTwinsCriterion
  from vissl.losses.simclr_info_nce_loss import SimclrInfoNCECriterion
  from vissl.utils.test_utils import gpu_test, init_distributed_on_file, with_temp_files
label_desc: Use image node_id 2 for calling the TestSimClrCriterionOnGpu obj's underlying member method code with example usage: obj.test_gather_embeddings_word_size_1() without return types