code: def __init__(self, nclass: int, **kwargs):
      self.params = EasyDict(kwargs)
      self.nclass = nclass
  
imports: import os
  import pickle
  from typing import Optional, Dict, Iterable, Callable
  import jax
  import jax.numpy
  import numpy
  from absl import flags
  from tqdm import tqdm, trange
  import objax
  from examples.fixmatch.libml.data.core import DataSet
  from objax.util import EasyDict
label_desc: Use image node_id 1 to create a new TrainLoopFSL object from inherited base classes: objax with example: obj = TrainLoopFSL(nclass)