arcface-pytorch icon indicating copy to clipboard operation
arcface-pytorch copied to clipboard

train hang time and fire this error another time

Open ofathy1981 opened this issue 4 years ago • 0 comments

        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

ofathy1981 avatar Jul 06 '20 21:07 ofathy1981