We currently require the file names to be of the format:
python/model_runner.py: module_name = "{}_baseline_keras2".format(model_name)
python/model_runner.py: module_name = "{}_baseline_pytorch".format(model_name)
this can be an input from test.sh or one of cfg- files