Module train

Module train 

Source

Functionsยง

back_propagation
forward_pass
train