This library provides an abstraction to perform Model Versioning using Weight & Biases.

Last update: Jan 28, 2022

Overview

Description

This library provides an abstraction to perform Model Versioning using Weight & Biases.

Features

Version a new trained model
Promote a model to another stage (e.g production)

Example of usage

The following code snippet shows how to promote a newly trained model to the best_model on the validation set. First, it creates the model providing the checkpoint path, the artifact name and type, a description, and the metadata, where the metrics or any desired information is stored.

The comparision_type indicates how to compare the metrics between the two models. E.g if smaller, the smaller the metric from the new model the better

run = wandb.init(...)

# Create the desired artifact
artifact = versioner.create_artifact(
                            checkpoint='model.ckpt',
                            artifact_name='prueba',
                            artifact_type='model',
                            description='Prueba Wandb-MV',
                            metadata={
                                'val_metric': 78.0,
                                'test_metric': 0.0
                            })

# Promote the desired artifact to the 'best_model' tag
versioner.promote_model(new_model=artifact,
                        artifact_name='prueba',
                        artifact_type='model',
                        comparision_metric='val_metric',
                        promotion_alias='best_model',
                        comparision_type='smaller'
                       )

This code snippet shows how to promote a trained model to production after being validated on the test set. The already_deployed parameter indicates that the model is already logged, so it only needs to be updated.

versioner = Versioner(run)
versioner.promote_model(new_model=artifact,
                        artifact_name='prueba',
                        artifact_type='model',
                        comparision_metric='test_metric',
                        promotion_alias='production',
                        comparision_type='smaller',
                        already_deployed=True
                        )

Next features

Allow providing a custom comparision function to promote a model
...

This library provides an abstraction to perform Model Versioning using Weight & Biases.

Related tags

Overview

Description

Features

Example of usage

Next features

Owner

Hector Lopez Almazan

Super Resolution for images using deep learning.

Image Captioning using CNN and Transformers

Repository for open research on optimizers.

Deep learning library for solving differential equations and more

g2o: A General Framework for Graph Optimization

Code for paper Novel View Synthesis via Depth-guided Skip Connections

Efficient training of deep recommenders on cloud.

Neural Magic Eye: Learning to See and Understand the Scene Behind an Autostereogram, arXiv:2012.15692.

PyTorch code for Composing Partial Differential Equations with Physics-Aware Neural Networks

ESGD-M - A stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch

Optimizes image files by converting them to webp while also updating all references.

A python library to artfully visualize Factorio Blueprints and an interactive web demo for using it.

PyTorch implementation of Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolation.

MDMM - Learning multi-domain multi-modality I2I translation

DeeBERT: Dynamic Early Exiting for Accelerating BERT Inference

Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.

An executor that performs image segmentation on fashion items

Numenta published papers code and data

On the model-based stochastic value gradient for continuous reinforcement learning

Official PyTorch Implementation of Learning Self-Similarity in Space and Time as Generalized Motion for Video Action Recognition, ICCV 2021