Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

**Feat: Integrate Loss Graphs into Gradio Interface** #2241

Open
wants to merge 4 commits into
base: main
Choose a base branch
from

Conversation

Wernervanrun
Copy link

Pull Request Checklist

  • The PR has a proper title. Use Semantic Commit Messages.
  • Ready to be merged into the relevant branch.
  • The code submitted has been tested and runs successfully.

PR Type

  • New Feature: Add loss graph functionality for Gradio interface.

Description

This pull request introduces a new feature to enhance the Gradio interface by adding the ability to visualize loss graphs:

  1. Add Loss Graph Functionality for Gradio Interface
    • Implemented helper functions to retrieve and display loss graphs for different projects.
    • Integrated these graphs with Gradio, providing interactive tabs for users to view and switch between different loss graphs.
    • Handled project selection logic to ensure that the correct loss graphs are displayed based on the user's selection.
    • Added helper functions for generating and saving loss graphs from TensorBoard logs, which are then displayed within the Gradio interface.

Screenshot

Capture_01

Capture_02

**Body:**
* Implements helper functions to retrieve and display loss graphs for different projects.
* Integrates with Gradio to provide interactive tabs with loss graph images.
* Handles project selection and updates loss graphs accordingly.
* Includes helper functions for generating and saving loss graphs from TensorBoard logs
- Updated `extract_scalar_data` to handle NaN values in TensorBoard logs.
- If a scalar value is NaN, the method now falls back to the previous valid value.
- If no previous value is available, a default of 0.0 is used.
- This ensures continuity and robustness in the extracted scalar data.
…ault project name, also moved default_project_name logic into the get_projects function.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant