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

#covfefe

Browse Source
This commit is contained in:
Joseph Redmon
2017-06-01 20:31:13 -07:00
parent 1ef829e585
commit 56d69e73ab
87 changed files with 915 additions and 888 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
examples/classifier.c
View File

@@ -1,10 +1,5 @@
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/option_list.h"
#include "darknet/blas.h"
#include "darknet/classifier.h"
#include "darknet/cuda.h"
#include "darknet.h"
#include <sys/time.h>
#include <assert.h>
@@ -37,11 +32,7 @@ void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int *gpus,
#ifdef GPU
cuda_set_device(gpus[i]);
#endif
nets[i] = parse_network_cfg(cfgfile);
if(weightfile){
load_weights(&nets[i], weightfile);
}
if(clear) *nets[i].seen = 0;
nets[i] = load_network(cfgfile, weightfile, clear);
nets[i].learning_rate *= ngpus;
}
srand(time(0));