mnist-tvault-example

Sample repository for classifying MNIST dataset with tvault model registry.

Installing tvault

pip install tvault

Description

This repository contains example usage of tvault. We use ResNet-18 model to classify MNIST dataset. Since the purpose of the repository is to let you experience model registry, we set train epoch to 5. In order to run experiment:

After experiment end, model registry will be created using statement in train.py:

tags = {"language": "pytorch", "size": "0.5x", "learning_rate": learning_rate}

tvault.log_all(model, tags=tags, result=acc.item(), optimizer=optimizer).

Note that you can add your custom tags in tags and apply them to your model registry. Also, you can skip all other elements for logging except for model itself.

While model registry uses the commit hash, multiple registries with different settings can be made in single commit. In this case, model indexes are used to differentiate different model experiments.

Command line tools

Above command automatically generates model registry for your pytorch model. Your model registry (model related code, result, tag, metatdata ... ) will be saved in model_log directory.

tvault --find_flag

tvault find_flag option allows you to look up different expereiments with simple cli. find_flag offers three different ways of exploring results.

hash tvault --find_flag --condition hash --hash 2ba4adf shows all experiments with hash 2ba4adf. It will show all tags within the registry.

result tvault --find_flag --condition result --min 90 --max 100 shows all experiments with result value between 50 and 100. Note that the table is generated using model registry with most number of tags. Empty boxes mean that the tag has not been set for the registry.

tag tvault --find_flag --condition tag --tag_type size --tag 0.5x shows all experiments where the tag value is 0.5x for tag type size.

tvault --diff_flag

tvault diff_flag option allows you to look up difference of two models by specifying model hash and index. tvault automatically detects functions that are realated to model, which removes git diffs that are not related to model.

tvault --diff_flag --sha1 f407ed0 --index1 0 --sha2 737b47a --index2 0 provides the model difference between models in between two commits.