code: def __init__(self, input_size: int, distance_emb_size: int):
      super().__init__()
      self.ffnn = torch.nn.Sequential(
          torch.nn.Linear(input_size * 2 + 64, input_size),
          torch.nn.ReLU(),
          torch.nn.Dropout(0.3),
          torch.nn.Linear(input_size, 256),
          torch.nn.ReLU(),
          torch.nn.Dropout(0.3),
          torch.nn.Linear(256, 64),
      )
      self.conv = torch.nn.Sequential(
          torch.nn.Conv1d(64, 4, 3, 1, 1),
          torch.nn.Conv1d(4, 2, 3, 1, 1),
      )
      self.emb = torch.nn.Embedding(128, distance_emb_size)
  
file_path: stanza/stanza/models/coref/span_predictor.py
imports: from typing import List, Optional, Tuple
  from stanza.models.coref.const import Doc, Span
  import torch
label_desc: Use image node_id 1 to create a new SpanPredictor object from inherited base classes: torch.nn with example: obj = SpanPredictor(input_size, distance_emb_size)