- no weighInit in prj

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@600 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2019-10-29 07:07:02 +00:00
parent a061fdf0da
commit e254521759
3 changed files with 7 additions and 5 deletions
+1 -1
View File
@@ -1,5 +1,5 @@
CONFIG ?= release
SRCS := source/main.cpp source/Rbm.cpp source/Layer.cpp source/Stack.cpp
SRCS := source/main.cpp source/Rbm.cpp source/Layer.cpp source/Stack.cpp source/noise.c
LIBS := -larmadillo -ljsoncpp
+5 -2
View File
@@ -105,16 +105,19 @@ int main()
stack.save();
// Shake weights
stack.weightsInit(0.01);
stack.saveWeights();
#else
// Load project
stack.load();
// Load weights
stack.loadWeights();
#endif
// Train stack
stack.train(batch, 1000, 1000, &statusDisplay);
stack.weightsInit(0.01);
stack.train(batch, 100, 1000, &statusDisplay);
// Save weights
stack.saveWeights();
+1 -2
View File
@@ -19,8 +19,7 @@
"momentum" : 0.5,
"numGibbs" : 1,
"weightDecay" : 0
},
"weightInit" : 0.01
}
},
"weights_file" : "Layer.0.weights.dat"
}