saving weight files as binaries, hell yeah

This commit is contained in:
Joseph Redmon
2015-02-06 18:53:53 -08:00
parent bfffadc755
commit 2f62fe33c9
9 changed files with 147 additions and 47 deletions

View File

@ -36,14 +36,12 @@ connected_layer *make_connected_layer(int batch, int inputs, int outputs, ACTIVA
float scale = 1./sqrt(inputs);
//scale = .01;
for(i = 0; i < inputs*outputs; ++i){
layer->weights[i] = scale*rand_normal();
}
for(i = 0; i < outputs; ++i){
layer->biases[i] = scale;
// layer->biases[i] = 1;
}
#ifdef GPU