class_docstr: Duplex (chromosome + metaphase) ViT model for classification task.
class_docstr_tok: ["Duplex","(","chromosome","+","metaphase",")","ViT","model","for","classification","task","."]
code: def build_flax_model(self):
      return DuplexViT(
          num_classes=self.dataset_meta_data["num_classes"],
          patches=self.config.model.patches,
          hidden_size=self.config.model.hidden_size,
          encoder=self.config.model.encoder,
          dropout_rate=self.config.model.get("dropout_rate", 0.0),
          classifier=self.config.model.classifier,
          dtype="float32",
      )
  
file_path: scenic/scenic/projects/tasseo/duplex_vit.py
imports: from typing import Any
  import flax.linen
  import jax
  import jax.numpy
  import ml_collections
  from scenic.model_lib.base_models.classification_model import ClassificationModel
  from scenic.model_lib.layers import nn_layers
  from scenic.projects.baselines import vit
label_desc: Use image node_id 1 for calling the DuplexViTClassificationModel obj's underlying member method code with example usage: obj.build_flax_model() and returns: DuplexViT