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

How to download training logs to view in tensorflow? #725

Open
1 task done
the-it-weirdo opened this issue Jun 12, 2024 · 1 comment
Open
1 task done

How to download training logs to view in tensorflow? #725

the-it-weirdo opened this issue Jun 12, 2024 · 1 comment
Assignees
Labels
question A HUB question that does not involve a bug

Comments

@the-it-weirdo
Copy link

Search before asking

Question

Hello, I trained a model with my own dataset. I used cloud training. I can see the graphs in the Ultralytics Hub. However, I am unable to add any axis title to the graph. My searches tell me that I can add axis title using tensorboard. How do I access the tranining logs for the same? How can I download the logs to open with tensorboard?

Additional

No response

@the-it-weirdo the-it-weirdo added the question A HUB question that does not involve a bug label Jun 12, 2024
@sergiuwaxmann sergiuwaxmann self-assigned this Jun 13, 2024
@sergiuwaxmann
Copy link
Member

@the-it-weirdo Hello!

Unfortunately, I am not very familiar with TensorBoard but if it helps, you might be able to get the data you need from the PyTorch export.

Alternatively, you can get the metrics using Ultralytics HUB SDK:

from hub_sdk import HUBClient

# Initialize the HUBClient
credentials = {"api_key": "<API-KEY>"}
client = HUBClient(credentials)
print("Authenticated:", client.authenticated)

# Get model by ID
model = client.model("<MODEL-ID>")

# Get metrics
metrics = model.get_metrics()
print("Metrics:", metrics)

# Get weights
weight_url = model.get_weights_url("best")
print("Weight URL:", weight_url)

Make sure to replace <API-KEY> and <MODEL-ID> with your actual API key and model ID. This script will authenticate you, fetch the training metrics, and provide a direct link to download the best weights.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question A HUB question that does not involve a bug
Projects
None yet
Development

No branches or pull requests

2 participants