search

lzx1413 / PytorchSSD

pytorch version of SSD and it's enhanced methods such as RFBSSD,FSSD and RefineDet
MIT License
709 stars 237 forks source link

RefineDet320 #5

Open shwoo93 opened 6 years ago

shwoo93 commented 6 years ago

Hello thank you for sharing your codes

There is one issue in RefineDet320 training. The training stops after training epoch of 10. Even I tried several times, always the same phenomenon occurs. Any suggestion?

Thanks

lzx1413 commented 6 years ago

I have the same problem and try to solve it now. The dataloader always stops after the testing phase. Only testing the dataset at last can solve this problem temporarily.

shwoo93 commented 6 years ago

Thanks for your reply.

Another issue in your RefineDet320 code. The transfer connection block is implemented as following. x = F.relu(F.relu(l(u(x)),inplace=True)+t,inplace=True)

However, according to the paper, it should be as following. x = F.relu(l(F.relu(u(x)+t,inplace=True)),inplace=True)

lzx1413 commented 6 years ago

Thanks, I have not reproduced the result mentioned in the original paper. There must be some wrong with my code.

ghost commented 5 years ago

The test stage only supports one gpu.