Skip to content

Instantly share code, notes, and snippets.

@armhold
Created June 3, 2016 16:23
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save armhold/58f55ddcc5ff15459679813b0c11c57e to your computer and use it in GitHub Desktop.
Save armhold/58f55ddcc5ff15459679813b0c11c57e to your computer and use it in GitHub Desktop.
code from the python notebook for training george_net
from cs231n.classifiers.convnet import *
#lr = 10 ** -3.45855
lr = 10 ** -3.2
reg = 10 ** -4.92902
weight_scale = 10 ** -1.62151
model = GeorgeNet(num_convnets=3, num_affine=3,
hidden_dim=500, # default: 500
weight_scale=weight_scale,
num_filters=64,
use_batchnorm=True,
reg=reg)
solver = Solver(model, data,
num_epochs=10, batch_size=50,
update_rule='adam',
optim_config={
'learning_rate': lr,
'lr_decay': 0.95,
},
verbose=True, print_every=20)
solver.train()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment