```Traceback (most recent call last):


 "bert_kashgari.py", line 21, in <module>

model.fit(train_x, train_y, valid_x, valid_y)


 "C:\Users\Urvish\Envs\tf_1_14\lib\site-packages\kashgari\tasks\base_model.py", line 310, in fit

**fit_kwargs)


 "C:\Users\Urvish\Envs\tf_1_14\lib\site-packages\tensorflow\python\keras\engine\training.py", line 1433, in fit_generator

steps_name='steps_per_epoch')


 "C:\Users\Urvish\Envs\tf_1_14\lib\site-packages\tensorflow\python\keras\engine\training_generator.py", line 264, in model_iteration

batch_outs = batch_function(*batch_data)


 "C:\Users\Urvish\Envs\tf_1_14\lib\site-packages\tensorflow\python\keras\engine\training.py", line 1153, in train_on_batch

extract_tensors_from_dataset=True)


 "C:\Users\Urvish\Envs\tf_1_14\lib\site-packages\tensorflow\python\keras\engine\training.py", line 2692, in _standardize_user_data

y, self._feed_loss_fns, feed_output_shapes)


 "C:\Users\Urvish\Envs\tf_1_14\lib\site-packages\tensorflow\python\keras\engine\training_utils.py", line 549, in check_loss_and_target_compatibility

' while using as loss `' + loss_name + '`. '
ValueError: A target array with shape (64, 31) was passed for output of shape (None, 100, 31) while using as loss `categorical_crossentropy`. This loss expects targets to have the same shape as the output.```

Traceback (most recent call last):


 "trainMLAgents.py", line 86, in <module>

tc.start_learning(env, trainer_config)


 "D:\Repos\AnimalAI-Olympics\venv\lib\site-packages\animalai_train\trainers\trainer_controller.py", line 217, in start_learning

new_info = self.take_step(env, curr_info)


 "D:\Repos\AnimalAI-Olympics\venv\lib\site-packages\animalai_train\trainers\trainer_controller.py", line 288, in take_step

trainer.update_policy()


 "D:\Repos\AnimalAI-Olympics\venv\lib\site-packages\animalai_train\trainers\ppo\trainer.py", line 343, in update_policy

run_out = self.policy.update(buffer.make_mini_batch(start, end), n_sequences)


 "D:\Repos\AnimalAI-Olympics\venv\lib\site-packages\animalai_train\trainers\buffer.py", line 197, in make_mini_batch

mini_batch[key] = np.array(self[key][start:end])
ValueError: could not broadcast input array from shape (129) into shape (1)
