uoip / SSD-variants

PyTorch implementation of several SSD based object detection algorithms.
MIT License
240 stars 56 forks source link

How could I train the DSOD ? #10

Open liguilan1227 opened 4 years ago

liguilan1227 commented 4 years ago

When I train the DSOD, I did not load the reducefc.pth, and I replaced the "SSD" with the "DSOD", but something wrong occured."IndexError: invalid index of a 0-dim tensor. Use tensor.item() in Python or tensor.item<T>() in C++ to convert a 0-dim tensor to a number " could you tell me what should I do to solve it?