diff --git a/README.md b/README.md index 9ba380e6..430dfe4d 100644 --- a/README.md +++ b/README.md @@ -57,6 +57,19 @@ us](mailto:edgeml@microsoft.com). to this [project](https://www.microsoft.com/en-us/research/project/resource-efficient-ml-for-the-edge-and-endpoint-iot-devices/). +If you use the EdgeML library in your projects or publications, please use the +following BibTeX to cite us. + +``` + @online{edgemlcode, + title = {{The EdgeML Library: An ML library for machine learning on the Edge}}, + author={Kusupati, Aditya and Dennis, Don and Gupta, Chirag and Kumar, Ashish and Patil, Shishir and Simhadri, Harsha}, + year = 2017, + url = {https://github.com/Microsoft/EdgeML} + } + +``` + ### Microsoft Open Source Code of Conduct This project has adopted the [Microsoft Open Source Code of diff --git a/pytorch/examples/ProtoNN/protoNN_example.ipynb b/pytorch/examples/ProtoNN/protoNN_example.ipynb index 38d6417c..cf3192f0 100644 --- a/pytorch/examples/ProtoNN/protoNN_example.ipynb +++ b/pytorch/examples/ProtoNN/protoNN_example.ipynb @@ -134,10 +134,11 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 6, "metadata": { "ExecuteTime": { - "start_time": "2019-06-30T10:21:37.236Z" + "end_time": "2019-06-30T10:25:13.443873Z", + "start_time": "2019-06-30T10:21:37.702352Z" }, "scrolled": true }, @@ -364,7 +365,14 @@ "Epoch 190 batch 0 loss 0.291277 acc 0.937500\n", "Epoch 191 batch 0 loss 0.291130 acc 0.937500\n", "Epoch 192 batch 0 loss 0.291667 acc 0.937500\n", - "Epoch 193 batch 0 loss 0.291187 acc 0.937500\n" + "Epoch 193 batch 0 loss 0.291187 acc 0.937500\n", + "Epoch 194 batch 0 loss 0.290773 acc 0.937500\n", + "Epoch 195 batch 0 loss 0.290442 acc 0.937500\n", + "Epoch 196 batch 0 loss 0.290126 acc 0.937500\n", + "Epoch 197 batch 0 loss 0.289970 acc 0.937500\n", + "Epoch 198 batch 0 loss 0.289415 acc 0.937500\n", + "Epoch 199 batch 0 loss 0.289136 acc 0.937500\n", + "Validation accuracy: 0.913802\n" ] } ], @@ -385,13 +393,26 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 7, "metadata": { "ExecuteTime": { - "start_time": "2019-06-30T10:21:37.237Z" + "end_time": "2019-06-30T10:25:13.469355Z", + "start_time": "2019-06-30T10:25:13.447362Z" } }, - "outputs": [], + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Final test accuracy tensor(0.9138, dtype=torch.float64)\n", + "Model size constraint (Bytes): 78240\n", + "Number of non-zeros: 19560\n", + "Actual model size: 78240\n", + "Actual non-zeros: 16368\n" + ] + } + ], "source": [ " x_, y_= torch.Tensor(x_test), torch.Tensor(y_test)\n", "logits = protoNNObj.forward(x_)\n",