training_environment

class brainpy.math.training_environment(dt=None, x64=None, complex_=None, float_=None, int_=None, bool_=None, batch_size=1, membrane_scaling=None, bp_object_as_pytree=None, numpy_func_return=None)

Environment with the training mode.

This is a short-cut context setting for an environment with the training mode. It is equivalent to:

>>> import brainpy.math as bm
>>> with bm.environment(mode=bm.training_mode):
>>>   pass