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

Minor fix

Browse Source
This commit is contained in:
AlexeyAB
2019-02-07 14:47:43 +03:00
parent c999f53e9d
commit 9bb7455a0e
5 changed files with 50 additions and 231 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/classifier.c
View File

@ -125,6 +125,8 @@ void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int *gpus,
load_thread = load_data(args);
int iter_save = get_current_batch(net);
int iter_save_last = get_current_batch(net);
while(get_current_batch(net) < net.max_batches || net.max_batches == 0){
time=clock();
@ -155,13 +157,23 @@ void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int *gpus,
draw_train_loss(img, img_size, avg_loss, max_img_loss, i, net.max_batches, -1, 0, dont_show, mjpeg_port);
#endif // OPENCV
if (i >= (iter_save + 100)) {
if (i >= (iter_save + 1000)) {
iter_save = i;
#ifdef GPU
if (ngpus != 1) sync_nets(nets, ngpus, 0);
#endif
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);
}
if (i >= (iter_save_last + 100)) {
iter_save_last = i;
#ifdef GPU
if (ngpus != 1) sync_nets(nets, ngpus, 0);
#endif
char buff[256];
sprintf(buff, "%s/%s_last.weights", backup_directory, base, i);
save_weights(net, buff);
}
free_data(train);

2
src/cuda.h
View File

@ -16,7 +16,7 @@ extern int gpu_index;
#include "curand.h"
#include "cublas_v2.h"
#include "cuda_runtime_api.h"
#include "driver_types.h"
//#include "driver_types.h"
#ifdef CUDNN
#include "cudnn.h"