Add option to bypass training and do benchmark on a provided model