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

Using LSTM write Tang poetry

The implementation of the CVPR2021 paper "Structure-Aware Face Clustering on a Large-Scale Graph with 10^7 Nodes"

[AAAI 2021] MVFNet: Multi-View Fusion Network for Efficient Video Recognition

[CVPR'22] COAP: Learning Compositional Occupancy of People

GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Models

A novel method to tune language models. Codes and datasets for paper ``GPT understands, too''.

Boostcamp CV Serving For Python

Unsupervised Pre-training for Person Re-identification (LUPerson)

Split your patch similarly to `git add -p` but supporting multiple buckets

[NeurIPS 2021] Large Scale Learning on Non-Homophilous Graphs: New Benchmarks and Strong Simple Methods

PyTorch implementation of Advantage async actor-critic Algorithms (A3C) in PyTorch

a reccurrent neural netowrk that when trained on a peice of text and fed a starting prompt will write its on 250 character text using LSTM layers

The official codes of "Semi-supervised Models are Strong Unsupervised Domain Adaptation Learners".

A simple, high level, easy-to-use open source Computer Vision library for Python.

Pytorch library for end-to-end transformer models training and serving

PyTorch implementation of Neural Dual Contouring.

Official Implementation of Neural Splines

Code for our SIGCOMM'21 paper "Network Planning with Deep Reinforcement Learning".

Diabetes-Feature-Engineering - A machine learning model that can predict whether people have diabetes when their characteristics are specified

In this project, two programs can help you take full agvantage of time on the model training with a remote server