Gradient computation error about efficientdet.pytorch HOT 2 OPEN

Hello,
I solved this
In the models directory, you should change lines 178 and 180 of the bifpn.py module.

Just replace this

w1 /= torch.sum(w1, dim=0) + self.eps  # normalize

w2 /= torch.sum(w2, dim=0) + self.eps  # normalize

to this:

w1 = w1/(torch.sum(w1, dim=0) + self.eps)  # normalize

w2 = w2/(torch.sum(w2, dim=0) + self.eps)  # normalize

from efficientdet.pytorch.

met this issue too, is there any solutions?

from efficientdet.pytorch.