Thanks! That makes sense based on the error message I’m seeing where the value error is showing slicing the 0th index of shape:
ValueError: Input dimension mismatch: (input[1].shape[0] = 45, input[4].shape[0] = 10000)
Still hitting some convergence issues but making progress