ramon-oliveira / aorun

Deep Learning over PyTorch

Aorun: Deep Learning over PyTorch

Aorun intend to implement an API similar to Keras with PyTorch as backend.

Getting started

Here is a simple regression example:

from aorun.models import Model
from aorun.layers import Dense, Activation

model = Model()
model.add(Dense(10, input_dim=3))
model.add(Activation('relu'))
model.add(Dense(1))

model.fit(X_train, y_train, loss='mse', optimizer='adam')

y_pred = model.predict(X_test)

Install

First of all, it's import to mention that this project is develop with Python 3.5+ in mind. I do not recommend using Aorun with older versions.

As prerequisite, you have to install the latest stable version of PyTorch

Then you can install aorun as any other python package, with pip:

$ pip install aorun

Why Aorun?

Aorun is a Dinosaur. Dinosaurs are cool.

aorun's Issues

MNIST assertion error

(py3) MBP:examples $ python mnist.py
11222it [00:04, 2786.04it/s]
(60000, 28, 28) (10000, 28, 28)
Traceback (most recent call last):
  File "mnist.py", line 38, in <module>
    history = model.fit(X, y, loss=loss, val_data=(X_test, y_test))
  File "/Users/MY/Dropbox (Personal)/x/aorun/aorun/models.py", line 92, in fit
    self._build()
  File "/Users/MY/Dropbox (Personal)/x/aorun/aorun/models.py", line 33, in _build
    next_layer.build(prev_layer.output_dim)
  File "/Users/MY/Dropbox (Personal)/x/aorun/aorun/layers.py", line 57, in build
    assert type(input_dim) is int
AssertionError

models.py

  File "aorun/aorun/models.py", line 39
    n_samples, *_ = X.size()
               ^
SyntaxError: invalid syntax```