From 549cbf9b3c94f71565ce88924861d741a748d8a2 Mon Sep 17 00:00:00 2001
From: Hammouda Elbez <hammouda.elbez@univ-lille.fr>
Date: Sun, 27 Nov 2022 10:57:30 +0000
Subject: [PATCH] FMNIST class updated
---
CSNN-Simulator/apps/FMnist.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/CSNN-Simulator/apps/FMnist.cpp b/CSNN-Simulator/apps/FMnist.cpp
index 0a848ac..00e61c1 100644
--- a/CSNN-Simulator/apps/FMnist.cpp
+++ b/CSNN-Simulator/apps/FMnist.cpp
@@ -41,8 +41,8 @@ int main(int argc, char** argv) {
std::string input_path(input_path_ptr);
- experiment.template add_train<dataset::FMnist>(input_path+"train-images-idx3-ubyte", input_path+"train-labels-idx1-ubyte", 1000);
- experiment.template add_test<dataset::FMnist>(input_path+"t10k-images-idx3-ubyte", input_path+"t10k-labels-idx1-ubyte", 1000);
+ experiment.template add_train<dataset::FMnist>(input_path+"train-images-idx3-ubyte", input_path+"train-labels-idx1-ubyte");
+ experiment.template add_test<dataset::FMnist>(input_path+"t10k-images-idx3-ubyte", input_path+"t10k-labels-idx1-ubyte");
float th_lr = 0.0f;
float t_obj = 0.75;
--
GitLab