class_docstr: Mixin for filtering the document containing the label out from retrieved docs.
  
  As these models are generally used with a CC knowledge dump, and the task on which
  they are evaluated is from the same CC dump, it is useful to filter out the document
  containing the label string from the retrieved documents, for a fair comparison.
class_docstr_tok: ["Mixin","for","filtering","the","document","containing","the","label","out","from","retrieved","docs",".","As","these","models","are","generally","used","with","a","CC","knowledge","dump",",","and","the","task","on","which","they","are","evaluated","is","from","the","same","CC","dump",",","it","is","useful","to","filter","out","the","document","containing","the","label","string","from","the","retrieved","documents",",","for","a","fair","comparison","."]
docstr: Cache the label vec.
docstr_tok: ["Cache","the","label","vec","."]
code: def set_labels(self, labels: torch.Tensor):
      """
      Cache the label vec.
      """
      self.label_vec = labels
  
imports: from typing import Optional, Type, Tuple, List, Dict, Any
  import torch
  import torch.nn
  from parlai.agents.fid.fid import FidModel, concat_enc_outs
  from parlai.agents.hugging_face.gpt2 import GPT2Decoder, Gpt2Agent
  from parlai.agents.rag.args import RetrieverType
  from parlai.agents.rag.modules import RagModel
  from parlai.agents.rag.retrievers import Document, BLANK_DOC, SearchQuerySearchEngineRetriever, retriever_factory, RagRetriever
  from parlai.core.dict import DictionaryAgent
  from parlai.core.opt import Opt
  import parlai.utils.logging
  from projects.seeker.agents.seeker_modules import ComboFidSearchQuerySearchEngineRetriever
  from projects.seeker.agents.seeker_modules import combo_fid_retriever_factory, interleave_fid_combo_outputs, ComboFidModel
label_desc: Use image node_id 1 for calling the FilterDocsForLabelSearchEngineRetrieverMixin obj's underlying member method code with example usage: obj.set_labels(labels) without return types