code: def get_all_heads(dim_in: int = 1024) -> nn.Module:
      heads = nn.ModuleDict(
          {
              "image": get_imagenet_head(dim_in),
              "rgbd": get_sunrgbd_head(dim_in),
              "video": get_kinetics_head(dim_in),
          }
      )
      return heads
  
imports: from towhee.models.video_swin_transformer.video_swin_transformer import VideoSwinTransformer
  from typing import Any, Optional, Union
  import torch
  from torch import nn
  from torch.hub import load_state_dict_from_url
label_desc: Use image node_id 1 for calling a global function with example usage: get_all_heads(dim_in) and returns: heads