- 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 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
View File
@@ -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();
+1 -2
View File
@@ -19,8 +19,7 @@
"momentum" : 0.5, "momentum" : 0.5,
"numGibbs" : 1, "numGibbs" : 1,
"weightDecay" : 0 "weightDecay" : 0
}, }
"weightInit" : 0.01
}, },
"weights_file" : "Layer.0.weights.dat" "weights_file" : "Layer.0.weights.dat"
} }