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

Pytorch implementation for DFN: Distributed Feedback Network for Single-Image Deraining.

[NeurIPS'20] Self-supervised Co-Training for Video Representation Learning. Tengda Han, Weidi Xie, Andrew Zisserman.

Rl-quickstart - Reinforcement Learning Quickstart

An end-to-end machine learning library to directly optimize AUC loss

Fast and robust clustering of point clouds generated with a Velodyne sensor.

A graph adversarial learning toolbox based on PyTorch and DGL.

MANO hand model porting for the GraspIt simulator

GNN4Traffic - This is the repository for the collection of Graph Neural Network for Traffic Forecasting

HyperCube: Implicit Field Representations of Voxelized 3D Models

Fake videos detection by tracing the source using video hashing retrieval.

EMNLP 2021: Single-dataset Experts for Multi-dataset Question-Answering

Liecasadi - liecasadi implements Lie groups operation written in CasADi

Official implementation of the PICASO: Permutation-Invariant Cascaded Attentional Set Operator

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

Lacmus is a cross-platform application that helps to find people who are lost in the forest using computer vision and neural networks.

MegEngine implementation of YOLOX

Implementation of Memory-Efficient Neural Networks with Multi-Level Generation, ICCV 2021

Source code for models described in the paper "AudioCLIP: Extending CLIP to Image, Text and Audio" (https://arxiv.org/abs/2106.13043)

ArcaneGAN by Alex Spirin

Official implementation of VQ-Diffusion