Comments (5)
change gpu mumber in the file rfcn_end2end_ohem_multi_gpu.sh ,the source code is 8 gpu
ime ./tools/train_net_multi_gpu.py --gpu ${GPU_ID}
--solver models/${PT_DIR}/${NET}/rfcn_end2end/solver_ohem.prototxt
--weights data/imagenet_models/${NET}-model.caffemodel
--imdb ${TRAIN_IMDB}
--iters ${ITERS}
--cfg experiments/cfgs/rfcn_end2end_ohem_${PT_DIR}.yml
${EXTRA_ARGS}
from py-r-fcn-multigpu.
HI Xiangyu, have you solved the problem? I have the same problem as you.
from py-r-fcn-multigpu.
@Xiangyu-CAS Have you solved the problem? I have the same problem as you.
from py-r-fcn-multigpu.
has anyone installed the caffe in this repo successfully?
from py-r-fcn-multigpu.
You have to export the path to the NCCL build. I made a local install so i exported it to my home
export LD_LIBRARY_PATH=/home/[username]/nccl/build/lib:$LD_LIBRARY_PATH
from py-r-fcn-multigpu.
Related Issues (20)
- Can't parse message of type "caffe.NetParameter" error HOT 1
- The error when using coco dataset
- How to train on multiple gpus?
- when use the ZF model, not the VGG16. The error happens
- GPU memory not release after interrupted the training script HOT 3
- Provide Resnet-152 training network scripts
- what is the difference between master branch and the coco branch?
- Cannot allocate memory
- How to download imagenet pretrained model HOT 1
- Can't parse message of type "caffe.NetParameter"
- Problem of multi-GPU "'NCCL' has no attribute 'new_uid'" HOT 1
- Is that possible to totally control the randomness? HOT 1
- how to solve the overflow encountered runtimewanring ?
- How to use this OHEM for Faster R-CNN?
- about coco branch, about Results on MS-COCO
- caffe version
- caffe version
- Train on coco got empty gt_box and crash
- VOC resnet50 error prototxt
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from py-r-fcn-multigpu.