Home

Awesome

Neptune

A unified codebase, programmed in Python 3, for executing a collection of neural networks being developed at Federal University of Rio Grande by one Computer Engineering student and a Ph.D student:

Networks

In the codebase there are three supported applications that run on neural networks:

Dependencies

Currently, our codebase depends on the following packages:

Execution

To choose the neural network and dataset combination to use, the file 'main.py' needs to be edited. If you need to change default configuration, such as shape or batch size, then you will also want to edit 'config.py'.

After setting up the appropriate networks, datasets and configurations, execution is as simple as:

$ python general.py

The command for tensorboard will be automatically generated and printed by the program in the terminal.

Configuration can also be specified via command-line arguments. For a list of all possibilities take a look at 'config.py', an example is provided below:

$ python general.py --batch_size=16 --learning_rate=1e-4

Publications

Below is a list of papers we published that are related to this codebase:

Datasets