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

View File

@ -89,7 +89,7 @@ void train_coco(char *cfgfile, char *weightfile)
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);
if(i%1000==0){
if(i%1000==0 || (i < 1000 && i%100 == 0)){
char buff[256];
sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
save_weights(net, buff);