Add argparse to davide_trainer