Currently, we have two args that always have to be identical (otherwise, training will fail). We should remove one of those args.