- 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:
@@ -1,5 +1,5 @@
|
|||||||
CONFIG ?= release
|
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
|
LIBS := -larmadillo -ljsoncpp
|
||||||
|
|||||||
+5
-2
@@ -105,16 +105,19 @@ int main()
|
|||||||
stack.save();
|
stack.save();
|
||||||
|
|
||||||
// Shake weights
|
// Shake weights
|
||||||
stack.weightsInit(0.01);
|
stack.saveWeights();
|
||||||
|
|
||||||
#else
|
#else
|
||||||
// Load project
|
// Load project
|
||||||
stack.load();
|
stack.load();
|
||||||
|
|
||||||
// Load weights
|
// Load weights
|
||||||
stack.loadWeights();
|
stack.loadWeights();
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Train stack
|
// Train stack
|
||||||
stack.train(batch, 1000, 1000, &statusDisplay);
|
stack.weightsInit(0.01);
|
||||||
|
stack.train(batch, 100, 1000, &statusDisplay);
|
||||||
|
|
||||||
// Save weights
|
// Save weights
|
||||||
stack.saveWeights();
|
stack.saveWeights();
|
||||||
|
|||||||
Reference in New Issue
Block a user