Added state parameters to pycaffe