Add gumbel-softmax for classification

[gomezzz]

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

  NIDN/nidn/training/model/model_to_eps_grid.py

  Line 199 in be21eeb

  def _classification_model_to_eps_grid(model, run_cfg: DotMap):

• Add a cfg flag to select

How Can It Be Tested

Run one of the notebooks using the classification with it.