Model Details

Model Description

This repo includes every models we trained during the Jax Community event sprint, organized by Hugging Face. The folders {model} contains the Flax checkpoint and {model}_pt the Torch checkpoint.

Uses

<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->

See our gradio app for more information : UCDR-Net gradio

Training Details

Training Data

Training Procedure

We trained from scratch each one of our models. We kept the initial parameters, except for the Batch Size. You can find the training script in the following Event repo's folder

Preprocessing

-Resize to 128 resolution -Canny Edge Map

Training parameters

The following table describes the differents hyperpa alt text

We stopped the coyo model a bit after it processed its first epoch. After running it, we discovered it performed pretty well even after only one epoch. So we deciced to keep it.

The last model has been trained with a custom DataLoader. The previous loads a batch containing 4 images from Bridge and 28 from Coyo. Therefore, we can't talk about epoch as the model processed coyo faster than bridge. We then trained the model according to steps and not epoch.

Results

See UCDR-Net gradio

Environmental Impact

<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->

Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).