Module

class ivy.neural_net_stateful.module.Module(dev_str=None, v=None)[source]

Bases: abc.ABC

__init__(dev_str=None, v=None)[source]

Initialze Ivy layer, which is a stateful object consisting of trainable variables.

Parameters
  • dev_str (str, optional) – device on which to create the layer’s variables ‘cuda:0’, ‘cuda:1’, ‘cpu’ etc.

  • v (ivy container, optional) – Ivy container of trainable variables. Created internally by default.

save_weights(weights_path)[source]

Save the weights on the Module. :param weights_path: The hdf5 file for saving the weights. :type weights_path: string


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty