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

Decoding the Protein-ligand Interactions Using Parallel Graph Neural Networks

Official PyTorch implementation of "Improving Face Recognition with Large AgeGaps by Learning to Distinguish Children" (BMVC 2021)

Code for ICCV 2021 paper: ARAPReg: An As-Rigid-As Possible Regularization Loss for Learning Deformable Shape Generators..

Single-Stage Instance Shadow Detection with Bidirectional Relation Learning (CVPR 2021 Oral)

[ICCV 2021] Deep Hough Voting for Robust Global Registration

Code for Mining the Benefits of Two-stage and One-stage HOI Detection

The Body Part Regression (BPR) model translates the anatomy in a radiologic volume into a machine-interpretable form.

CTRL-C: Camera calibration TRansformer with Line-Classification

World Models with TensorFlow 2

GPU-accelerated PyTorch implementation of Zero-shot User Intent Detection via Capsule Neural Networks

Torch implementation of SegNet and deconvolutional network

Implementation of Sequence Generative Adversarial Nets with Policy Gradient

Official git for "CTAB-GAN: Effective Table Data Synthesizing"

Multi-Scale Progressive Fusion Network for Single Image Deraining

Using multidimensional LSTM neural networks to create a forecast for Bitcoin price

The code for our paper submitted to RAL/IROS 2022: OverlapTransformer: An Efficient and Rotation-Invariant Transformer Network for LiDAR-Based Place Recognition.

Learning Lightweight Low-Light Enhancement Network using Pseudo Well-Exposed Images

A PyTorch implementation of the Transformer model in "Attention is All You Need".

An efficient 3D semantic segmentation framework for Urban-scale point clouds like SensatUrban, Campus3D, etc.

pytorch, hand(object) detect ,yolo v5，手检测