@@ -18,9 +18,11 @@ Read the paper [here](https://arxiv.org/abs/1902.06714).
18
18
19
19
* Dense, fully connected neural layers
20
20
* Convolutional and max-pooling layers (experimental, forward propagation only)
21
+ * Flatten layers (forward and backward pass)
22
+ * Loading dense and convolutional models from Keras h5 files
21
23
* Stochastic and mini-batch gradient descent for back-propagation
22
24
* Data-based parallelism
23
- * Several activation functions
25
+ * Several activation functions and their derivatives
24
26
25
27
### Available layer types
26
28
@@ -57,7 +59,8 @@ Required dependencies are:
57
59
Optional dependencies are:
58
60
59
61
* OpenCoarrays (for parallel execution with GFortran)
60
- * BLAS, MKL (optional)
62
+ * BLAS, MKL, or similar (for offloading ` matmul ` and ` dot_product ` calls)
63
+ * curl (for downloading testing and example datasets)
61
64
62
65
Compilers tested include:
63
66
@@ -200,13 +203,15 @@ examples, in increasing level of complexity:
200
203
dataset
201
204
4 . [ cnn] ( example/cnn.f90 ) : Creating and running forward a simple CNN using
202
205
` input ` , ` conv2d ` , ` maxpool2d ` , ` flatten ` , and ` dense ` layers.
203
- 5 . [ mnist_from_keras] ( example/mnist_from_keras.f90 ) : Creating a pre-trained
204
- model from a Keras HDF5 file.
206
+ 5 . [ dense_from_keras] ( example/dense_from_keras.f90 ) : Creating a pre-trained
207
+ dense model from a Keras HDF5 file and running the inference.
208
+ 6 . [ cnn_from_keras] ( example/cnn_from_keras.f90 ) : Creating a pre-trained
209
+ convolutional model from a Keras HDF5 file and running the inference.
205
210
206
211
The examples also show you the extent of the public API that's meant to be
207
212
used in applications, i.e. anything from the ` nf ` module.
208
213
209
- The MNIST example uses [ curl] ( https://curl.se/ ) to download the dataset ,
214
+ Examples 3-6 rely on [ curl] ( https://curl.se/ ) to download the needed datasets ,
210
215
so make sure you have it installed on your system.
211
216
Most Linux OSs have it out of the box.
212
217
The dataset will be downloaded only the first time you run the example in any
0 commit comments