Code

Add mixed precision support#46

Code

Merged

julienperichon merged 6 commits intomasterClementWalter/Keras-FewShotLearning:masterfrom add-mixed-precision-supportClementWalter/Keras-FewShotLearning:add-mixed-precision-supportCopy head branch name to clipboardJul 29, 2022

+206-804

Commits

Commits on Jul 28, 2022

• Add .python-version to .gitignore for pyenv environments

  julienperichon

  committed
  3602a3f

  Copy full SHA for 3602a3f

• Cast ClippedBinaryCrossEntropy parameters to same type as loss for mixed precision support

  julienperichon

  committed
  b162a6d

  Copy full SHA for b162a6d

Commits on Jul 29, 2022

• Set activation layer of LearntNorms to be of float32 dtype policy if global policy is mixed precision

  julienperichon

  committed
  6252b65

  Copy full SHA for 6252b65

• Add ClippedBinaryCrossEntropy test for all float dtypes

  julienperichon

  committed
  18a6434

  Copy full SHA for 18a6434

• Update CI config to install correct version of Tensorflow

  julienperichon

  committed
  79ec9c7

  Copy full SHA for 79ec9c7

• Remove activation in last Dense layer

  julienperichon

  committed
  4c67235

  Copy full SHA for 4c67235