Lightning-AI/pytorch-lightning

CLI for inspecting checkpoints

Open

#5.551 geöffnet am 18. Jan. 2021

Auf GitHub ansehen
 (2 Kommentare) (7 Reaktionen) (1 zugewiesene Person)Python (3.233 Forks)batch import
checkpointingdesignfeaturehelp wanted

Repository-Metriken

Stars
 (26.687 Stars)
PR-Merge-Metriken
 (Durchschn. Merge 9T 15h) (3 gemergte PRs in 30 T)

Beschreibung

🚀 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

Contributor Guide