arcface-pytorch
arcface-pytorch copied to clipboard
train hang time and fire this error another time
data_input, label = data
data_input = data_input.to(device)
label = label.to(device).long()
Traceback (most recent call last):
File "train.py", line 177, in
any help with this error