code: def main():
      # Load the experiment. Replace the run uuid with you run uuid from
      # [training the model](train_model.html).
      conf = load_experiment("4984b85c20bf11eb877a69c1a03717cd")
      # Set model to evaluation mode
      conf.model.eval()
  
      # Collect $\big(f(c_i), w_i\big)$
      gather_keys(conf)
      # Add them to the index for fast search
      build_index(conf)
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/transformers/knn/build_index.py
imports: from typing import Optional
  import faiss
  import numpy
  import torch
  from labml import experiment, monit, lab
  from labml.utils.pytorch import get_modules
  from labml_nn.transformers.knn.train_model import Configs
label_desc: Use image node_id 4 for calling a global function with example usage: main() without return types