Skip to content

CLI for inspecting checkpoints #5551

@awaelchli

Description

@awaelchli

🚀 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

Metadata

Metadata

Assignees

Labels

checkpointingRelated to checkpointingdesignIncludes a design discussionfeatureIs an improvement or enhancementhelp wantedOpen to be worked on

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions