Comments (5)
do you have the code for compress the yolo-v3 model, as yolo-v3 model with 416&416 need GPU memory 2~3GB, so do you have any suggest for reduce the memory?
you want to reduce the memory used in training or just test process? If you want to reduce it in forward you can commit the calloc function of updates parameters such as l.weight_updates_gpu, l.delta_gpu
another suggest is combine the convolution and batch normalization into one function, you can also save some memory
from yolov3_lite.
can you give some detail explain about the commit calloc function, as i just want reduce in test stage on device, not training
from yolov3_lite.
can you give some detail explain about the commit calloc function, as i just want reduce in test stage on device, not training
e-mail: [email protected], send me a
can you give some detail explain about the commit calloc function, as i just want reduce in test stage on device, not training
in the start of program, it will call make_convolutional_layer function, and in this function will calloc gpu memory according to your parameters, but this calloc are not all necessary for test process, so you can commit them :
l.delta_gpu = cuda_make_array(l.delta, l.batchout_hout_w*n);
if you have further question, you can send me an e-mail: [email protected]
from yolov3_lite.
@wangnet Where did you find the pruned weights file for YOLO V2?
from yolov3_lite.
Hello, I am working on a custom dataset using the tiny yolov3 architecture, I already used the AlexeyAB git repo, I got results but I need to embed prunning as well, shall I use the standard cfg file and weights file on your framework? Or there are extra steps I shall take before starting the data training?
from yolov3_lite.
Related Issues (10)
- can the model file after training (just use google quantization) be correctly parsed by original darknet
- L1 scales regularization HOT 2
- Weights File for yolov2 HOT 10
- Predictions are not drawing after testing trained model HOT 8
- Usage of this repo HOT 2
- Is this implementation for YOLOV2 or YOLOV3? HOT 7
- `Hash Compress` `Google Quantization` `Huffman Compress`
- AVX Problem HOT 6
- how to get output_scale and output_zero_point in the cfg file.
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 yolov3_lite.