There is someyhing error ,and I changed logits,_, _= net(images) into logits, cc, aa= net(images) as it used to make error,but now it has the same error about racnn-pytorch

[jeong-tae]

can you down the batch size more smaller?

[deepblue0822]

ok,I will try,thank you

[jeong-tae]

I can't catch up How reproduce that error. Normally, StopIteration error is occur with loader. If you change the dataloader, plz carefully check with it first. I am not recommend to use pretrainAPN because it is not fully implemented. Just comment out pretrainAPN part and try again. Please don't reply this message in the e-mail. Instead of mailing, plz use git issue dashboard to get support from others also. Thanks Jeongtae, Lee

…

-----Original Message----- From: "deepblue0822"<[email protected]> To: "jeong-tae/RACNN-pytorch"<[email protected]>; Cc: "jeong-tae"<[email protected]>; "Comment"<[email protected]>; Sent: 2018-06-29 (금) 12:32:33 Subject: Re: [jeong-tae/RACNN-pytorch] There is someyhing error ,and I changed logits,_, _= net(images) into logits, cc, aa= net(images) as it used to make error,but now it has the same error (#3) when I down the batchsize to 2 ,and the error is Traceback (most recent call last): File "trainer2.py", line 310, in train() File "trainer2.py", line 61, in train apn_iter, apn_epoch, apn_steps = pretrainAPN(trainset, trainloader) File "trainer2.py", line 206, in pretrainAPN images, labels = next(batch_iterator) File "/usr/local/lib/python3.5/dist-packages/torch/utils/data/dataloader.py", line 276, in next raise StopIteration StopIteration — You are receiving this because you commented. Reply to this email directly, view it on GitHub, or mute the thread.

[deepblue0822]

OK,Thank you very much.I tried down the batch_size and it worked but the accuracy rate is very low,and I think maybe my memory is not enough.And I will try to do something to improve it.Whatever thank you very much.

[jeong-tae]

I hope you get good results. Even in my experiments, i can't load more than batch size 4.
If you find any error or improvements, let me know and share it please.