dibgerge / blog-tf-mxnet-pytorch

Tensorflow, PyTorch, and MxNet

An experimental and subjective comparison

This is the code used for the blog here.

Run the file frameworks_compare.py to generate the error and timing results for tensorflow, pytorch and mxnet.