Lightning-AI/pytorch-lightning

CLI for inspecting checkpoints

Open

#5,551 opened on 2021幎1月18日

GitHub で芋る
 (2 comments) (7 reactions) (1 assignee)Python (3,233 forks)batch import
checkpointingdesignfeaturehelp wanted

Repository metrics

Stars
 (26,687 stars)
PR merge metrics
 (平均マヌゞ 9d 15h) (30d で 3 merged PRs)

説明

🚀 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

コントリビュヌタヌガむド