esa / esa/NIDN

Add gumbel-softmax for classification

Open
#116 0 comments 0 reactions 0 assignees View on GitHub
enhancement help wanted
Dominant language
Jupyter Notebook
Stars
54
Forks
12
PR merge metrics
No merged PRs in 30d

Description

# Feature
## Desired Behavior / Functionality
An alternative to the currently utilized approach of building a linear combination of materials may be the use of a gumbel-softmax as described here https://arxiv.org/abs/1611.01144 .

## What Needs to Be Done
- Add gumbel softmax in network -> eps conversion in https://github.com/esa/NIDN/blob/be21eeb08b01782ee6c2735f193796c2b4bcbd2c/nidn/training/model/model_to_eps_grid.py#L199
- Add a cfg flag to select

## How Can It Be Tested
Run one of the notebooks using the classification with it.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.