"""Training LeNet on MNIST data."""

import athenet
from athenet.data_loader import MNISTDataLoader


network = athenet.models.lenet(trained=False)
network.data_loader = MNISTDataLoader()

config = athenet.TrainConfig()

config.n_epochs = 10
config.batch_size = 300
config.learning_rate = 0.1
config.val_interval = 1
config.val_units = 'epochs'

network.train(config)
print 'Accuracy on test data: {:.2f}%'.format(100*network.test_accuracy())

"""Training LeNet on MNIST data."""

import athenet
from athenet.data_loader import MNISTDataLoader


network = athenet.models.lenet(trained=False)
network.data_loader = MNISTDataLoader()

config = athenet.TrainConfig()

config.n_epochs = 10
config.batch_size = 300
config.learning_rate = 0.1
config.val_interval = 1
config.val_interval_units = 'epochs'

network.train(config)
print 'Accuracy on test data: {:.2f}%'.format(100*network.test_accuracy())

Fix name of config field
Fix name of config field
