hfsoftmax icon indicating copy to clipboard operation
hfsoftmax copied to clipboard

'torch.dtype' object has no attribute 'type'

Open Linsongrong opened this issue 4 years ago • 2 comments

@yl-1993 Hi, i try to train the model with webface, then this problem appears。 image How should I solve this problem

Linsongrong avatar Jul 13 '20 08:07 Linsongrong

我也遇到这个问题,请问你解决了吗

yjhuasheng avatar Nov 19 '20 08:11 yjhuasheng

utils.py中的
self.avg = np.mean(self.history) 改成
try: self.avg = torch.mean(torch.stack(self.history)) except: self.avg = np.mean(self.history)

xad1997 avatar Jun 09 '21 13:06 xad1997