brainpy.math.training_mode# brainpy.math.training_mode = TrainingMode(batch_size=1)# Training mode requires data batching.