Module#
- class Module(*args: Any, **kwargs: Any)#
Attributes
trainingstateMethods
applyapply_to_moduleschildrenevalfilter_and_mapfreezeleaf_modulesload_weightsmodulesnamed_modulesparameterssave_weightsset_dtypetraintrainable_parametersunfreezeupdateupdate_modules