Use monitoring_batches=1 to avoid class representation issues