config.lua |
A unified file for all configurations for the dataset, model, trainer, tester and GUI |
data.lua |
Provides a Data class; both training and validating datasets are instances of this class |
main.lua |
The main driver program |
model.lua |
Provides a Model class; it handles model creation, randomization and transformations during training |
mui.lua |
Provides a Mui class; uses the Scroll class to draw an nn.Sequential model in Qt |
scroll.lua |
Provides a Scroll class that starts a scrollable Qt window to draw text or images |
scroll.ui |
A Qt designer UI file corresponding to the scrollable Qt window |
test.lua |
Provides a Test class; handles testing, giving you losses, errors and confusion matrices |
train.lua |
Provides a Train class; handles training with Stochastic Gradient Descent (SGD) and supports things like momentum and weight decay |