google / google/tunix

grpo_demo.ipynb: Excessive output

Open
#687 4 comments 0 reactions 1 assignee Claimed by @selamw1 View on GitHub
type:feature/enhancement
Dominant language
Python
Stars
2.5k
Forks
345
Avg merge
1d 7h
Merged PRs (30d)
240

Description

This isn't an outright bug, just something that makes it difficult to work with `grpo_demo.ipynb`.

The cell with `grpo_trainer.train` produces a huge amount of text output. It becomes 95% of the notebook. This makes it difficult to scroll around the notebook or even tell when it's finished. I understand that you would like to show samples of outputs, but if it's important, I'd suggest showing a much smaller selection of outputs to keep the notebook workable.

If you agree with this I'll be happy to help.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.