elektronn2.neuromancer.graphmanager module

class elektronn2.neuromancer.graphmanager.GraphManager(name='')

Bases: object

function_code()

static get_lock_names(node_descriptors, count)

node_count

plot()

register_node(node, name, args, kwargs)

register_split(node, func, name, args, kwargs)

reset()

restore(descriptors, override_mfp_to_active=False, make_weights_constant=False, replace_bn=False, lock_trainable_params=False, unlock_all_params=False)

serialise()

sinks

sources