SV-X-Softmax usage error

cavalleria / cavaface

face recognition training project(pytorch)

MIT License

456 stars 88 forks source link

SV-X-Softmax usage error #2

Closed xsacha closed 4 years ago

xsacha commented 4 years ago

When training with SV-X-Softmax, the following error occurs:

 File "/home/imagus/dev/cavaface.pytorch/head/metrics.py", line 452, in forward
    if self.xtype == 'MV-AM':
  File "/home/imagus/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 594, in __getattr__
    type(self).__name__, name))
AttributeError: 'SVXSoftmax' object has no attribute 'xtype'

cavalleria commented 4 years ago

Sorry for the late reply, the bug has been fixed