class_docstr: Implements the Bit ResNet model for multi-label classification.
class_docstr_tok: ["Implements","the","Bit","ResNet","model","for","multi-label","classification","."]
code: def build_flax_model(self) -> nn.Module:
      return BitResNet(
          num_outputs=self.dataset_meta_data["num_classes"],
          gn_num_groups=self.config.get("gn_num_groups", 32),
          width_factor=self.config.get("width_factor", 1),
          num_layers=self.config.num_layers,
      )
  
imports: from typing import Any, Dict, Optional, Sequence, Tuple, Union
  from absl import logging
  import flax
  import flax.linen
  import jax.numpy
  import ml_collections
  from scenic.common_lib import debug_utils
  from scenic.model_lib.base_models.classification_model import ClassificationModel
  from scenic.model_lib.base_models.multilabel_classification_model import MultiLabelClassificationModel
  from scenic.model_lib.layers import nn_layers
label_desc: Use image node_id 1 for calling the BitResNetMultiLabelClassificationModel obj's underlying member method code with example usage: obj.build_flax_model() and returns: BitResNet