code: def create_pred_layers(self):
      return {
          "heatmap": self.create_pred_layer(1),
          "offset": self.create_pred_layer(2),
      }
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from backbones.upsample_head import SimpleUpsampleHead
label_desc: Use image node_id 2 for calling the SimpleMSRDecoder obj's underlying member method code with example usage: obj.create_pred_layers() and returns: dict