search

hyseob / PyMDNet

MDNet PyTorch implementation
Other
421 stars 150 forks source link

Runtime Error: from negative region generation #9

Open Peratham opened 7 years ago

Peratham commented 7 years ago

Hi Hyeonseob,

I found py-MDNet very interesting and I would like to use it a lot. So, I tried to train a new model based on a new dataset.

However, when I was running the program for the 4th cycle, it returns "RuntimeError: the given numpy array has zero-sized dimensions. Zero-sized dimensions are not supported in PyTorch".

It is from data_prov.py on line 63 neg_regions = torch.from_numpy(neg_regions).float().

called by train_mdnet.py on line 64 pos_regions, neg_regions = dataset[k].next()

Thank you. Best, Peratham

Cosmopal commented 5 years ago

Getting the same issue. For positive regions though...