Romil Shah

icon indicating a website link www.shahromil.com

icon company Amazon Web Services icon location USA icon location MS student at Northeastern University. Major in Computer Vision, Machine Learning & Algorithms. Proficient in Python, MATLAB, C++ & R.

Results 1 comments of Romil Shah

performance regression? caffe -> caffe2

@raininglixinyu Assign CPU and GPU separately for the different layers. Try something implemented by KeyKy here: [https://github.com/KeyKy/caffe2/blob/master/caffe2/python/tutorials/Run_Alexnet_in_CPU_and_GPU_mode.ipynb](https://github.com/KeyKy/caffe2/blob/master/caffe2/python/tutorials/Run_Alexnet_in_CPU_and_GPU_mode.ipynb)