Better partial function

This commit is contained in:
Joseph Redmon
2015-07-20 16:16:26 -07:00
parent 23c08be144
commit 38bd6ae6ba
3 changed files with 8 additions and 3 deletions

View File

@ -36,7 +36,7 @@ void partial(char *cfgfile, char *weightfile, char *outfile, int max)
load_weights_upto(&net, weightfile, max);
}
net.seen = 0;
save_weights(net, outfile);
save_weights_upto(net, outfile, max);
}
#include "convolutional_layer.h"