search

qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
MIT License
4.73k stars 1.03k forks source link

Added hardcoded float32 for the activation layer #536

Closed romitjain closed 1 year ago

romitjain commented 2 years ago

Added hardcoding of float32 for the activation layer.

This will prevent overriding from the following call:

from tensorflow.keras import mixed_precision
mixed_precision.set_global_policy('mixed_float16')

This can help enable mixed precision training on Unet models