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

LSTC: Boosting Atomic Action Detection with Long-Short-Term Context

My 1st place solution at Kaggle Hotel-ID 2021

Neural network for stock price prediction

NeuroFind - A solution to the to the Task given by the Oberseminar of Messtechnik Institute of TU Dresden in 2021

The official code for PRIMER: Pyramid-based Masked Sentence Pre-training for Multi-document Summarization

Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.

Light-SERNet: A lightweight fully convolutional neural network for speech emotion recognition

Deep Watershed Transform for Instance Segmentation

Monitor your ML jobs on mobile devices📱, especially for Google Colab / Kaggle

[ICME 2021 Oral] CORE-Text: Improving Scene Text Detection with Contrastive Relational Reasoning

Bayesian-Torch is a library of neural network layers and utilities extending the core of PyTorch to enable the user to perform stochastic variational inference in Bayesian deep neural networks

The official implementation of our CVPR 2021 paper - Hybrid Rotation Averaging: A Fast and Robust Rotation Averaging Approach

Official repository for the paper "Self-Supervised Models are Continual Learners" (CVPR 2022)

RIFE - Real-Time Intermediate Flow Estimation for Video Frame Interpolation

FasterAI: A library to make smaller and faster models with FastAI.

An introduction to bioimage analysis - http://bioimagebook.github.io

Official pytorch implementation of "Feature Stylization and Domain-aware Contrastive Loss for Domain Generalization" ACMMM 2021 (Oral)

Safe Local Motion Planning with Self-Supervised Freespace Forecasting, CVPR 2021

Python Jupyter kernel using Poetry for reproducible notebooks

Modeling CNN layers activity with Gaussian mixture model