hfsoftmax
hfsoftmax copied to clipboard
'torch.dtype' object has no attribute 'type'
@yl-1993 Hi, i try to train the model with webface, then this problem appears。
How should I solve this problem
我也遇到这个问题,请问你解决了吗
utils.py中的
self.avg = np.mean(self.history)
改成
try:
self.avg = torch.mean(torch.stack(self.history))
except:
self.avg = np.mean(self.history)