code: def get_test_task_config(head_num_classes=1000):
      return {
          "name": "classification_task",
          "num_epochs": 12,
          "loss": {"name": "CrossEntropyLoss"},
          "dataset": {
              "train": {
                  "name": "synthetic_image",
                  "crop_size": 224,
                  "class_ratio": 0.5,
                  "num_samples": 2000,
                  "seed": 0,
                  "batchsize_per_replica": 32,
                  "num_workers": 1,
                  "use_shuffle": True,
                  "transforms": [
                      {
                          "name": "apply_transform_to_key",
                          "transforms": [
                              {"name": "ToTensor"},
                              {
                                  "name": "Normalize",
                                  "mean": [0.485, 0.456, 0.406],
                                  "std": [0.229, 0.224, 0.225],
                              },
                          ],
                          "key": "input",
                      }
                  ],
              },
              "test": {
                  "name": "synthetic_image",
                  "crop_size": 224,
                  "class_ratio": 0.5,
                  "num_samples": 2000,
                  "seed": 0,
                  "batchsize_per_replica": 32,
                  "num_workers": 1,
                  "use_shuffle": False,
                  "transforms": [
                      {
                          "name": "apply_transform_to_key",
                          "transforms": [
                              {"name": "ToTensor"},
                              {
                                  "name": "Normalize",
                                  "mean": [0.485, 0.456, 0.406],
                                  "std": [0.229, 0.224, 0.225],
                              },
                          ],
                          "key": "input",
                      }
                  ],
              },
          },
          "meters": {"accuracy": {"topk": [1, 5]}},
          "model": {
              "name": "resnet",
              "num_blocks": [3, 4, 6, 3],
              "small_input": False,
              "zero_init_bn_residuals": True,
              "heads": [
                  {
                      "name": "fully_connected",
                      "unique_id": "default_head",
                      "num_classes": head_num_classes,
                      "fork_block": "block3-2",
                      "in_plane": 2048,
                  }
              ],
          },
          "optimizer": {
              "name": "sgd",
              "num_epochs": 12,
              "param_schedulers": {
                  "lr": {"name": "step", "values": [0.1, 0.01]}
              },
              "weight_decay": 1e-4,
              "momentum": 0.9,
          },
      }
  
file_path: ClassyVision/test/generic/config_utils.py
imports: import sys
  from classy_vision.tasks import build_task
label_desc: Use image node_id 1 for calling a global function with example usage: get_test_task_config(head_num_classes) and returns: dict