Home

Awesome

Torch7-Network Profiler

Repository to calculate feed-forward time and number of operations taken by a neural network.

Running the application

The application can profile both a written definition of a network or an already trained network (saved as ether ascii or binary). Pass in the location of the model using the -m or its long equivalent --model.

To profile the written definition of the model it must be defined in a specially formatted table saved to a file with the .lua extension. Examples of which can be found in the models directory.

th profile-model.lua --model <'path/filename.lua'> --res 1x3x231x231

To profile the already trained network, pass in the path and file name by again using the -m/model flag. If the network file has standard extensions the application will auto detect if the network is saved as ether an ascii or binary network and load appropriately.

Profiling the network speed on different platforms is also possible. Currently the default platform is cpu but if available the profiler can be targeted to run the networks using cuda.

th profile-model.lua --model <'path/filename.lua'> --platform <'cpu'|'cuda'>

License

This software is released under a creative commons license which allows for personal and research use only. For a commercial license please contact the authors. You can view a license summary here: http://creativecommons.org/licenses/by-nc/4.0/