Save only best model in Trainer

My bad it works, it was an issue with how i passed params through sys.argv