Solved by NullPointer in post #2 
              
              
                I had more inputs than I was testing with. Changing the number of inputs fixed this but I’m not sure how it affected batch size.