Hello,thanks for your projects,when i use my own database something is wrong,the problems is:
Traceback (most recent call last):
File "main.py", line 213, in
main()
File "main.py", line 96, in main
train(model, criterion_xent, criterion_cent,optimizer_model, optimizer_centloss, train_loader, use_gpu, 2, epoch)
File "main.py", line 120, in train
for batch_idx, (data, labels) in enumerate(trainloader):
ValueError: too many values to unpack (expected 2)
Hello,thanks for your projects,when i use my own database something is wrong,the problems is: Traceback (most recent call last): File "main.py", line 213, in
main()
File "main.py", line 96, in main
train(model, criterion_xent, criterion_cent,optimizer_model, optimizer_centloss, train_loader, use_gpu, 2, epoch)
File "main.py", line 120, in train
for batch_idx, (data, labels) in enumerate(trainloader):
ValueError: too many values to unpack (expected 2)
I don't know how to fix it.thank you very much.