class_docstr: Combines all coref modules together to find coreferent spans.
  
  Attributes:
      config (coref.config.Config): the model's configuration,
          see config.toml for the details
      epochs_trained (int): number of epochs the model has been trained for
      trainable (Dict[str, torch.nn.Module]): trainable submodules with their
          names used as keys
      training (bool): used to toggle train/eval modes
  
  Submodules (in the order of their usage in the pipeline):
      tokenizer (transformers.AutoTokenizer)
      bert (transformers.AutoModel)
      we (WordEncoder)
      rough_scorer (RoughScorer)
      pw (PairwiseEncoder)
      a_scorer (AnaphoricityScorer)
      sp (SpanPredictor)
class_docstr_tok: ["Combines","all","coref","modules","together","to","find","coreferent","spans",".","Attributes",":","config","(","coref.config.Config",")",":","the","model","'s","configuration",",","see","config.toml","for","the","details","epochs_trained","(","int",")",":","number","of","epochs","the","model","has","been","trained","for","trainable","(","Dict","[","str",",","torch.nn.Module","]",")",":","trainable","submodules","with","their","names","used","as","keys","training","(","bool",")",":","used","to","toggle","train\/eval","modes","Submodules","(","in","the","order","of","their","usage","in","the","pipeline",")",":","tokenizer","(","transformers.AutoTokenizer",")","bert","(","transformers.AutoModel",")","we","(","WordEncoder",")","rough_scorer","(","RoughScorer",")","pw","(","PairwiseEncoder",")","a_scorer","(","AnaphoricityScorer",")","sp","(","SpanPredictor",")"]
code: def _set_training(self, value: bool):
      self._training = value
      for module in self.trainable.values():
          module.train(self._training)
  
imports: from datetime import datetime
  import json
  import logging
  import os
  import random
  import re
  from typing import Any, Dict, List, Optional, Set, Tuple
  import numpy
  import toml
  import torch
  import transformers
  from stanza.models.coref import bert, conll, utils
  from stanza.models.coref.anaphoricity_scorer import AnaphoricityScorer
  from stanza.models.coref.cluster_checker import ClusterChecker
  from stanza.models.coref.config import Config
  from stanza.models.coref.const import CorefResult, Doc
  from stanza.models.coref.loss import CorefLoss
  from stanza.models.coref.pairwise_encoder import PairwiseEncoder
  from stanza.models.coref.rough_scorer import RoughScorer
  from stanza.models.coref.span_predictor import SpanPredictor
  from stanza.models.coref.tokenizer_customization import TOKENIZER_FILTERS, TOKENIZER_MAPS
  from stanza.models.coref.utils import GraphNode
  from stanza.models.coref.word_encoder import WordEncoder
  from peft import LoraConfig, get_peft_model, get_peft_model_state_dict, set_peft_model_state_dict
  from stanza.utils.get_tqdm import get_tqdm
label_desc: Use image node_id 20 for calling the CorefModel obj's underlying member method code with example usage: obj._set_training(value) without return types