[ADD] option to cli for retrain network