fork download

copy

embedding_norm = F.normalize(x)*self.s
weight_norm = F.normalize(self.weight)
zy = F.linear(embedding_norm, weight_norm)
cosine_t = zy/self.s
sine_t = torch.sqrt(1.0 - cosine_t*cosine_t)
cos_phi = (cosine_t * self.cos_m - sine_t * self.sin_m)*self.s
 
if self.easy_margin:
    cos_phi = torch.where(cosine_t > 0, cos_phi, zy)
else:
    cos_phi = torch.where(cosine_t > self.threshold, cos_phi, zy - self.s * self.mm)
 
 # --------------------------- convert label to one-hot ---------------------------
one_hot = torch.zeros(cosine_t.size(), device='cuda')
# one_hot = one_hot.cuda() if cosine.is_cuda else one_hot
one_hot.scatter_(1, label.view(-1, 1).long(), 1)
output = (one_hot * cos_phi) + ((1.0 - one_hot) * zy)
#output = one_hot*(cos_phi-zy) + zy
return output

Traceback (most recent call last):
  File "/usr/lib/python3.7/py_compile.py", line 143, in compile
    _optimize=optimize)
  File "<frozen importlib._bootstrap_external>", line 791, in source_to_code
  File "<frozen importlib._bootstrap>", line 219, in _call_with_frames_removed
  File "./prog.py", line 19
    return output
    ^
SyntaxError: 'return' outside function

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "/usr/lib/python3.7/py_compile.py", line 147, in compile
    raise py_exc
py_compile.PyCompileError:   File "./prog.py", line 19
    return output
                ^
SyntaxError: 'return' outside function

Standard output is empty