Skip a test in debug mode that is too slow.

Ideally we should run the script for just one epoch and make the model smaller. How can we do this? Can you make in a yaml file a variable with default value?