brainpy.math.training_mode#

brainpy.math.training_mode = TrainingMode(batch_size=1)#

Training mode requires data batching.