KL在计算LOSS的问题

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.

https://rotationdetection.readthedocs.io/

Apache License 2.0

1.08k stars 181 forks source link

Closed lalalagogogochong closed 2 years ago

lalalagogogochong commented 2 years ago

在计算loss时，矩阵相乘之后，我复现的时候会出现虚数的问题，pytorch 在求导时只能对实数进行求导的，针对这个虚数问题，你们有碰到过吗，处理的时候是怎么处理呢，tf可以解决这个问题吗