mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

idk

Browse Source
This commit is contained in:
Joseph Redmon 2016-06-25 16:13:54 -07:00
parent cf32e7e9b8
commit a6b2511a56
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Makefile
View File

@ -1,6 +1,6 @@
GPU=0 GPU=0
OPENCV=0
CUDNN=0 CUDNN=0
OPENCV=0
DEBUG=0 DEBUG=0
ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52 ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52

4
cfg/yolo-coco.cfg
View File

@ -1,6 +1,6 @@
[net] [net]
batch=1 batch=64
subdivisions=1 subdivisions=4
height=448 height=448
width=448 width=448
channels=3 channels=3

2
src/coco.c
View File

@ -89,7 +89,7 @@ void train_coco(char *cfgfile, char *weightfile)
avg_loss = avg_loss*.9 + loss*.1; avg_loss = avg_loss*.9 + loss*.1;
printf("%d: %f, %f avg, %f rate, %lf seconds, %d images\n", i, loss, avg_loss, get_current_rate(net), sec(clock()-time), i*imgs); printf("%d: %f, %f avg, %f rate, %lf seconds, %d images\n", i, loss, avg_loss, get_current_rate(net), sec(clock()-time), i*imgs);
if(i%1000==0){ if(i%1000==0 || (i < 1000 && i%100 == 0)){
char buff[256]; char buff[256];
sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i); sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
save_weights(net, buff); save_weights(net, buff);