Module

Contents

Module#

class Module(*args: Any, **kwargs: Any)#

Attributes

training

state

Methods

apply

apply_to_modules

children

eval

filter_and_map

freeze

leaf_modules

load_weights

modules

named_modules

parameters

save_weights

set_dtype

train

trainable_parameters

unfreeze

update

update_modules