Custom metric

Added support for a custom metric. Changed the tutorial run.py so it can use multiple gpus