code: def _create_distillation_config(
      checkpoint_path: str, num_gpu: int = 2
  ):
      with initialize_config_module(config_module="vissl.config"):
          cfg = compose(
              "defaults",
              overrides=[
                  "config=debugging/distillation/distill_rn50_to_rn50_mse_trunk",
                  f"config.DISTILLATION.TEACHER_MODEL.WEIGHTS_INIT.PARAMS_FILE={checkpoint_path}",
                  "config.TEST_MODEL=False",
                  "config.DATA.TRAIN.DATA_SOURCES=[synthetic]",
                  "config.DATA.TRAIN.LABEL_SOURCES=[synthetic]",
                  "config.DATA.TRAIN.LABEL_TYPE=sample_index",
                  "config.DATA.TRAIN.DATA_LIMIT=40",
                  "config.DATA.TRAIN.BATCHSIZE_PER_REPLICA=4",
                  "config.SEED_VALUE=0",
                  "config.DISTRIBUTED.NUM_NODES=1",
                  f"config.DISTRIBUTED.NUM_PROC_PER_NODE={num_gpu}",
                  "config.LOG_FREQUENCY=1",
                  "config.OPTIMIZER.num_epochs=2",
              ],
          )
      args, config = convert_to_attrdict(cfg)
      return config
  
file_path: vissl/tests/test_distillation.py
imports: import os
  import unittest
  from hydra.experimental import compose, initialize_config_module
  from vissl.utils.hydra_config import convert_to_attrdict
  from vissl.utils.test_utils import gpu_test, in_temporary_directory, run_integration_test
label_desc: Use image node_id 2 for calling the TestDistillation obj's underlying member method code with example usage: obj._create_distillation_config(checkpoint_path, num_gpu) and returns: config