train hang time and fire this error another time

ronghuaiyang / arcface-pytorch

1.74k stars 392 forks source link

train hang time and fire this error another time #70

Open ofathy1981 opened 4 years ago

ofathy1981 commented 4 years ago

        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 data_input = data_input.to(device) AttributeError: 'list' object has no attribute 'to'

any help with this error