Lightning-AI/pytorch-lightning

CLI for inspecting checkpoints

Open

#5 551 ouverte le 18 janv. 2021

Voir sur GitHub
 (2 commentaires) (7 réactions) (1 assigné)Python (3 233 forks)batch import
checkpointingdesignfeaturehelp wanted

Métriques du dépôt

Stars
 (26 687 stars)
Métriques de merge PR
 (Merge moyen 9j 15h) (3 PRs mergées en 30 j)

Description

🚀 Feature

A quick way to inspect checkpoint files.

Motivation

We don't know what's inside a checkpoint. Loading one and checking the contents requires us to write a python script. A command line tool could help us load and inspect these checkpoint files conveniently. Useful for debugging too.

Pitch

My implementation is here. Docs

Usage:

> peek path/to/checkpoint.ckpt
displays all top-level variables in a checkpoint file


> peek -i path/to/checkpoint.ckpt
interactive mode, loads the contents in a python interactive shell and let's you modify the contents

peek

Let me know if you want it

cc @borda @tchaton @justusschock @awaelchli @ananthsub @ninginthecloud @rohitgr7

Guide contributeur