pytorch-center-loss icon indicating copy to clipboard operation
pytorch-center-loss copied to clipboard

about use own database

Open jiangyanjavawin opened this issue 5 years ago • 0 comments

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.

jiangyanjavawin avatar Feb 14 '20 14:02 jiangyanjavawin