diff --git a/CSNN-Simulator/apps/MnistForSpiNNaker.cpp b/CSNN-Simulator/apps/MnistForSpiNNaker.cpp index 65ed956e4557cea4032000306c5683030eb1c010..8513c89dc9a011454075e2c82ebee3967987b9be 100644 --- a/CSNN-Simulator/apps/MnistForSpiNNaker.cpp +++ b/CSNN-Simulator/apps/MnistForSpiNNaker.cpp @@ -34,7 +34,7 @@ int main(int argc, char** argv) { experiment.template add_test<dataset::Mnist>(input_path+"t10k-images.idx3-ubyte", input_path+"t10k-labels.idx1-ubyte"); float th_lr = 1.0f; - float t_obj = 0.50f; + float t_obj = 0.75f; float alpha = 0.05f; float alpha_p= 0.01f; float alpha_n= 0.005f; diff --git a/CSNN-Simulator/src/layer/Convolution.cpp b/CSNN-Simulator/src/layer/Convolution.cpp index 245c3559ee1f22e194155a8ba152388cb80d2952..b4efb5d80ed9833b167b70acab6e698887c11395 100644 --- a/CSNN-Simulator/src/layer/Convolution.cpp +++ b/CSNN-Simulator/src/layer/Convolution.cpp @@ -257,6 +257,7 @@ void _priv::DenseImpl::train(const std::vector<Spike>& input_spike, const Tensor _a.at(0, 0, z) += w.at(spike.x, spike.y, spike.z, z); // check if the voltage crossed the threshold if(_a.at(0, 0, z) >= th.at(z)) { + _model.layer_Spikes+=1; for(size_t z1=0; z1<depth; z1++) { th.at(z1) -= _model._lr_th*(spike.time - _model._t_obj); @@ -321,4 +322,4 @@ void _priv::DenseImpl::test(const std::vector<Spike>& input_spike, const Tensor< } } } -} \ No newline at end of file +}