code: def get_test_mlp_task_config():
      return {
          "name": "classification_task",
          "num_epochs": 10,
          "loss": {"name": "CrossEntropyLoss"},
          "dataset": {
              "train": {
                  "name": "synthetic_image",
                  "num_classes": 2,
                  "crop_size": 20,
                  "class_ratio": 0.5,
                  "num_samples": 20,
                  "seed": 0,
                  "batchsize_per_replica": 6,
                  "use_augmentation": False,
                  "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",
                  "num_classes": 2,
                  "crop_size": 20,
                  "class_ratio": 0.5,
                  "num_samples": 10,
                  "seed": 0,
                  "batchsize_per_replica": 1,
                  "use_augmentation": False,
                  "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",
                      }
                  ],
              },
          },
          "model": {
              "name": "mlp",
              # 3x20x20 = 1200
              "input_dim": 1200,
              "output_dim": 1000,
              "hidden_dims": [10],
              "use_batchnorm": True,  # used for testing sync batchnorm
          },
          "meters": {"accuracy": {"topk": [1]}},
          "optimizer": {
              "name": "sgd",
              "num_epochs": 10,
              "lr": 0.1,
              "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 4 for calling a global function with example usage: get_test_mlp_task_config() and returns: dict