]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
mnist : update README
authorGeorgi Gerganov <redacted>
Thu, 13 Apr 2023 21:02:31 +0000 (00:02 +0300)
committerGitHub <redacted>
Thu, 13 Apr 2023 21:02:31 +0000 (00:02 +0300)
examples/mnist/README.md

index 6b34470555354b5a8665f21b58a3093c508d27fa..bb08888407fdf8eba9edda9ec580cbfd3586883b 100644 (file)
@@ -17,9 +17,9 @@ plus the model weights and biases. Run convert-h5-to-ggml.py to convert your pyt
 
 - magic constant (int32)
 - repeated list of tensors
--- number of dimensions of tensor (int32)
--- tensor dimension (int32 repeated)
--- values of tensor (int32)
+- number of dimensions of tensor (int32)
+- tensor dimension (int32 repeated)
+- values of tensor (int32)
 
 Run ```convert-h5-to-ggml.py mnist_model.state_dict``` where `mnist_model.state_dict` is the saved pytorch model from the Google Colab. For
 quickstart, it is included in the mnist/models directory.
@@ -43,6 +43,7 @@ make -j4 mnist
 
 # Run the MNIST model
 ./bin/mnist ../examples/mnist/models/mnist/ggml-model-f32.bin ../examples/mnist/models/mnist/t10k-images.idx3-ubyte
+```
 
 For more information, checkout the corresponding programs in the [examples](examples) folder.
 
@@ -85,3 +86,7 @@ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _
 ggml_graph_dump_dot: dot -Tpng mnist.dot -o mnist.dot.png && open mnist.dot.png
 Predicted digit is 9
 ```
+
+Computation graph:
+
+![mnist dot](https://user-images.githubusercontent.com/1991296/231882071-84e29d53-b226-4d73-bdc2-5bd6dcb7efd1.png)