predict_geolocation.ipynb errors

[ranman]

https://github.com/multimedia-berkeley/tutorials/blob/master/predict_geolocation.ipynb

On the line: sym, arg_params, aux_params = mx.model.load_checkpoint(prefix, 2)
This fails to find the "2" param file.
I assume we're meant to use:
sym, arg_params, aux_params = mx.model.load_checkpoint(prefix, 12)