Make channel sizes and batch sizes adjustable