NameError: name 'long' is not defined

[pengchengma]

When I run this program , an error occurs:

"NameError: name 'long' is not defined"

it points to line 11 of file focalloss.py:

if isinstance(alpha,(float,int,long)): self.alpha = torch.Tensor([alpha,1-alpha])

[clcarwin]

The code work in python2, python3 do not has long type. You can delete all code lines include "alpha". I will rewrite code include "alpha" these days. It is not as same as the paper.
Thanks.

[Songxinlei]

what is updated code?

[snoop2head]

I just deleted the long argument from isinstance() which still works!

Instead of if isinstance(alpha,(float,int,long)): self.alpha = torch.Tensor([alpha,1-alpha]),

I used if isinstance(alpha,(float,int)): self.alpha = torch.Tensor([alpha,1-alpha])