OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Ubintu 18.04
TensorFlow version and how it was installed (source or binary): 2.3.0
TensorFlow-Addons version and how it was installed (source or binary): 0.12.0
Python version: 3.8.5
Is GPU used? (yes/no): both
Describe the bug
When using tfa.losses.SigmoidFocalCrossEntropy(from_logits=True, reduction='auto') the model raised ValueError:
ValueError: Found unexpected keys that do not correspond to any Model output: dict_keys(['class_name', 'config']). Expected: ['dense']
System information
Describe the bug
When using
tfa.losses.SigmoidFocalCrossEntropy(from_logits=True, reduction='auto')
the model raised ValueError:ValueError: Found unexpected keys that do not correspond to any Model output: dict_keys(['class_name', 'config']). Expected: ['dense']
Code to reproduce the issue
Other info / logs