Update generation utilities #172
Conversation
|
@PhungVanDuy Are we adding best-of-n here? |
Yes, we can add it here. |
|
Are we planning to merge tonight |
|
No, I still have to update every other example (currently grappling with T5's) and make regression plots. I will also add clarifying comments shortly so until don't rush with reviewing |
| # Log and display evaluation metrics | ||
| if self.accelerator.is_main_process: | ||
| rows = sum(list(map(list, zip(*table))), []) | ||
| rich_table = Table(*columns, title=f"Evaluation #{self.nth_evaluation}") |
There was a problem hiding this comment.
It's so beautiful 😭 Do you think we should add show_lines=True in the Table constructor? I noticed some of the outputs kind of bleed together but this might just be my terminal settings. Fine either way - much better than before!
| self, prompts: List[torch.IntTensor], samples, prompt_sizes=None | ||
| ) -> List[str]: | ||
| """ | ||
| Decode samples into (samples: List[str], outputs: List[str], samples: List[str]) |
There was a problem hiding this comment.
Should the return be documented instead as:
(samples: List[str], prompts: List[str], outputs: List[str])
trlx/trainer/__init__.py
Outdated
| config: TRLConfig, | ||
| reward_fn=None, | ||
| metric_fn=None, | ||
| stop_word=None, |
There was a problem hiding this comment.
In general, I don't believe that stop sequences will only be words. For clarity, either document that this handles a stop sequence (string) or update the arg name (e.g. OpenAI's Completion API calls it stop).
| stop_word_ix = str_output.find(self.stop_word) | ||
| if stop_word_ix == -1: | ||
| stop_word_ix = None | ||
| str_output = str_output[:stop_word_ix] |
There was a problem hiding this comment.
Maybe we want to right-strip this to avoid extra white space? (Otherwise, assume users expect to include any leading white space in their stop_word)
| for k, v in self.config.method.gen_kwargs.items(): | ||
| if isinstance(v, list): | ||
| if self.generate_sweep_kwarg is not None: | ||
| print( |
There was a problem hiding this comment.
Let's use utils.print_rank_0 or check for main process here to avoid the annoying prints from all processes.
This PR will
gen_kwargsvalue (e.g.temperature,top_k,beta) during periodic evaluationsreward_fn's signature fromreward_fn(samples)toreward_fn(samples, prompts, responses)gen_kwargs(e.g. Models indicate they have completed a response by generating a stop sequence, which is literally the string "Human:" – from Anthropic's 2021 "A General Language Assistant as a Laboratory for Alignment")Among HF's
generateargumentsStoppingCriteriacan only stop all generations per batch,eos_token_idsaccepts onlyList[int]. The other option is to customizegeneratemethod for every architecture or to trim samples aftergenerateas was done herehttps://wandb.ai/sorry/trlx/reports/-Update-generation-utilities-172---VmlldzozMzIxMjE1
https://wandb.ai/sorry/trlx/reports/Update-generation-utilities-172--VmlldzozMzIxMjE5