remove batch_size from models