class Problem:
- # returns a nb x (L+1+L) long tensor
- def generate_seq(self, nb):
+ def nb_token_values(self):
pass
- # save a file to vizualize quizzes, you can save a txt or png file
- def save_quizzes(self, input, result_dir, filename_prefix, logger):
+ # returns two tensors nb x D and nb x D'
+ def generate_prompts_and_answers(self, nb):
pass
- # returns a pair (forward_tokens, backward_token)
- def direction_tokens(self):
+ # save a file to vizualize quizzes, you can save a txt or png file
+ def save_quizzes(
+ self,
+ result_dir,
+ filename_prefix,
+ prompts,
+ answers,
+ predicted_prompt=None,
+ predicted_answers=None,
+ ):
pass