This probably happens when preparing the last batch. Add allow_smaller_final_batch=True
to your train.batch_join
invocation: from the documentation:
This probably happens when preparing the last batch. Add allow_smaller_final_batch=True
to your train.batch_join
invocation: from the documentation: