Skip to content
Snippets Groups Projects
Commit 549cbf9b authored by Hammouda Elbez's avatar Hammouda Elbez :computer:
Browse files

FMNIST class updated

parent 9c1608dc
No related branches found
No related tags found
No related merge requests found
......@@ -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;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment