save things sometimes

This commit is contained in:
Joseph Redmon 2015-09-22 16:40:51 -07:00
parent 522e2a220c
commit 2a1025b6f0
1 changed files with 1 additions and 3 deletions

View File

@ -74,9 +74,7 @@ void train_writing(char *cfgfile, char *weightfile)
epoch = *net.seen/N;
char buff[256];
sprintf(buff, "%s/%s_%d.weights",backup_directory,base, epoch);
//TODO
if(get_current_batch(net)%10 == 0)
save_weights(net, buff);
save_weights(net, buff);
}
}
}