YuwenXiong / py-R-FCN

R-FCN with joint training and python support
MIT License
1.05k stars 471 forks source link

How to set iterations when training with rfcn_alt_opt_5stage_ohem #124

Open wander1985 opened 5 years ago

wander1985 commented 5 years ago

I am trying to train my own data using rfcn_alt_opt_5stage_ohem and pascal_voc dataset. But I cannot find any parameters related to iterations. I can find ITERS=xxx in rfcn_end2end.sh and rfcn_end2end_ohem.sh. But in rfcn_alt_opt_5stage_ohem.sh, ITERS for pascal_voc dataset is nowhere to find. I have tried to add ITERS=100 in rfcn_alt_opt_5stage_ohem.sh line 30, but it just kept training after 100 iterations. Someone please help. Thanks.