setup to run sample training