<commit_before>import argparse

from model import Model

parser = argparse.ArgumentParser()
parser.add_argument('--datadir', required=True, dest='datadir')
parser.add_argument('--savefreq', default=10, dest='save_freq')
parsed = parser.parse_args()

model = Model(parsed.datadir, save_freq=parsed.save_freq, img_shape=(256, 448))
model.build_model()
model.build_losses()
model.build_optimizers()
model.setup_session()
model.setup_logging()
model.train()<commit_msg>batch size is command line option<commit_after>import argparse

from model import Model

parser = argparse.ArgumentParser()
parser.add_argument('--datadir', required=True, dest='datadir')
parser.add_argument('--savefreq', default=10, dest='save_freq')
parser.add_argument('--batchsize', default=64, dest='batch_size')
parsed = parser.parse_args()

model = Model(parsed.datadir, save_freq=parsed.save_freq, 
    batch_size=parsed.batch_size, img_shape=(256, 448))
model.build_model()
model.build_losses()
model.build_optimizers()
model.setup_session()
model.setup_logging()
model.train()
{'fraction_non_alphanumeric': 0.142721217887726, 'fraction_numerical': 0.017126546146527116, 'mean_word_length': 8.147826086956522, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5635098', 'n_tokens_mistral': 355, 'n_tokens_neox': 347, 'n_words': 62}