Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Learning in CPU #278

Open
marifnst opened this issue Jul 31, 2016 · 7 comments
Open

Learning in CPU #278

marifnst opened this issue Jul 31, 2016 · 7 comments

Comments

@marifnst
Copy link

dear all

i try to implement faster r cnn in CPU mode.
actually, faster r cnn officially doesnt provide training in CPU mode.

so, i made changes:

  • roi_pooling_layer.cpp -->conversion based on roi_pooling_layer.cu
  • smooth_L1_loss_layer.cpp --> conversion based on smooth_L1_loss_layer.cu

it's compiled successfully and training can be run smoothly using ZF network (my training process still in progress right now).
anybody can help to give suggestion about my changes on attached code ? is my code conversion correct ?

regards
muhammad arifn nasution

changes code.zip

@Austriker
Copy link

@marifnst Have a look at this PR. Some people are also trying to implement de CPU mode.

@marifnst
Copy link
Author

marifnst commented Aug 1, 2016

@Austriker thank you for your information. i will visit the thread.
my code running well and i intend to compare learning result in GPU mode.
because i intend to train faster r cnn in very low resource (ex mobile GPU)
will be shared soon.

@saiprabhakar
Copy link

@Austriker @marifnst I implemented the code based on GPU code of smooth l1 loss, but the gpu code doesnt take weights into consideration in its back propagation.
Is this a mistake or am I missing something?

@indsak
Copy link

indsak commented Jun 8, 2017

@marifnst Can you please tell me what all to be done after changing the files for training FRCNN in CPU mode?
Do i need to change the files or is it possible by giving any argument in the train option

@abhigoku10
Copy link

@marifnst hi tq for sharing the .cpp files is it possible to share the .cu files also since when i replace the existing files with urs i am ot able to get any changes

@sdews
Copy link

sdews commented Sep 13, 2017

@abhigoku10 After replaceing the existing files with the above two .cpp files , you should
cd $FRCN_ROOT/caffe-fast-rcnn
sudo make clean
sudo make -j $(($(nproc) + 1)) && make pycaffe

@Kshitij-Parashar
Copy link

Hi @marifnst,
I followed all the instructions and replaced both the files but could not get train in cpu mode.
Please help me, I would be highly greatful!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

7 participants