- 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
|
||||
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
@@ -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();
|
||||
|
||||
Reference in New Issue
Block a user