rxn.metrics.class_tokens.maybe_prepare_class_token_files

rxn.metrics.class_tokens.maybe_prepare_class_token_files(class_tokens, retro_files)[source]

If the model is a class-token one, create the expected src and target files.

Parameters
  • class_tokens (Optional[int]) – the number of tokens used in the trainings.

  • retro_files (RetroFiles) – information on the location of files for the metrics.

Return type

None