eruffaldi / paper_simulinkad

MNIST testing scripts for Tensorflow for the paper about Automatic Differentiation in Simulink

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

paper_simulinkad

Tensorflow

Tensorflow example was originally implemented for V1 with no Keras modules

Usage:

python mnist_tf_cnn.py python mnist_tf_softmax.py python mnist_tf_cifar.py

Pytorch

Single function for everything using --dataset mnist|cifar --model cnn|softmax

Common Settings

--batch-size number --optimizer adadelta|adam|gdescent --epochs number --lr learning-rate --gamma --seed

About

MNIST testing scripts for Tensorflow for the paper about Automatic Differentiation in Simulink

License:Apache License 2.0


Languages

Language:Python 100.0%