Issue: Using the extracted pre-trained style encoder results in output embeddings with NaN values, but only during training.
Resolution: Training was initially running on mixed precision at 16 bits. However, changing it to 32 bits resolved the issue, and only correct embeddings were generated.
Issue: Using the extracted pre-trained style encoder results in output embeddings with NaN values, but only during training.
Resolution: Training was initially running on mixed precision at 16 bits. However, changing it to 32 bits resolved the issue, and only correct embeddings were generated.