<s> from . . layers import Module <EOL> class Container ( Module ) : <EOL> def __init__ ( self , * modules ) : <EOL> Module . __init__ ( self ) <EOL> self . modules = [ ] <EOL> for module in modules : <EOL> self . add ( module ) <EOL> def evaluate ( self ) : <EOL> Module . evaluate ( self ) <EOL> for module in self . modules : <EOL> module . evaluate ( ) <EOL> def training ( self ) : <EOL> Module . training ( self ) <EOL> for module in self . modules : <EOL> module . training ( ) <EOL> def parameters ( self ) : <EOL> params , grads = [ ] , [ ] <EOL> for module in self . modules : <EOL> mod_params , mod_grads = module . parameters ( ) <EOL> params += mod_params <EOL> grads += mod_grads <EOL> return params , grads <EOL> def get_stat_updates ( self ) : <EOL> stat_updates = [ ] <EOL> for module in self . modules : <EOL> stat_updates += module . get_stat_updates ( ) <EOL> return stat_updates <EOL> def add ( self , module ) : <EOL> self . modules . append ( module ) </s>
