Skip to content

Instantly share code, notes, and snippets.

@juliensimon

juliensimon/LoadResnext101.py

Last active April 24, 2017 08:42
Show Gist options
  • Save juliensimon/452e3d83d09bac9cbc5e2b5ec2342c10 to your computer and use it in GitHub Desktop.
Save juliensimon/452e3d83d09bac9cbc5e2b5ec2342c10 to your computer and use it in GitHub Desktop.
Download ZIP
Load ResNext-101
Raw
LoadResnext101.py
import mxnet as mx
epochs = 100
sym, arg_params, aux_params = mx.model.load_checkpoint("resnext-101",0)
mod = mx.mod.Module(symbol=sym, context=(mx.gpu(0), mx.gpu(1), mx.gpu(2), mx.gpu(3)))
mod.bind(data_shapes=train_iter.provide_data, label_shapes=train_iter.provide_label)
mod.set_params(arg_params, aux_params)
mod.fit(train_iter, eval_data=valid_iter,
optimizer_params={'learning_rate':0.05, 'momentum':0.9}, num_epoch=epochs)
mod.save_checkpoint("resnext-101-001", epochs)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment