pytorch / pytorch/vision

Document other weight meta-data fields

Open
#5,982 0 comments 4 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

module: documentation
Dominant language
Python
Stars
17.9k
Forks
7.3k
Avg merge
1d 15h
Merged PRs (30d)
13

Description

We currently document the number of parameters in the Weights meta-data and we display these on our summary tables. Other values might be of interest for users:

CC @datumbox

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by tracing the existing Weight metadata and the summary-table generation, then compare the file-size documentation added in pull requests 6936 and 7074. Define how model size should be measured, especially for quantized models, and investigate torch.cuda.memory_stats for forward and backward memory estimates. Done means the selected measurements are computed consistently and documented for users.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
computer-vision, documentation
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.