MnistModel#
Attributes
This allows better BC support for |
Methods
|
Adds a child module to the current module. |
|
Applies |
Casts all floating point parameters and buffers to |
|
|
Returns an iterator over module buffers. |
Returns an iterator over immediate children modules. |
|
Moves all model parameters and buffers to the CPU. |
|
|
Moves all model parameters and buffers to the GPU. |
Casts all floating point parameters and buffers to |
|
Sets the module in evaluation mode. |
|
Set the extra representation of the module |
|
Casts all floating point parameters and buffers to |
|
Run a forward step on the network. |
|
|
Returns the buffer given by |
Returns any extra state to include in the module's state_dict. |
|
|
Returns the parameter given by |
|
Returns the submodule given by |
Casts all floating point parameters and buffers to |
|
|
Copies parameters and buffers from |
Returns an iterator over all modules in the network. |
|
|
Returns an iterator over module buffers, yielding both the name of the buffer as well as the buffer itself. |
Returns an iterator over immediate children modules, yielding both the name of the module as well as the module itself. |
|
|
Returns an iterator over all modules in the network, yielding both the name of the module as well as the module itself. |
|
Returns an iterator over module parameters, yielding both the name of the parameter as well as the parameter itself. |
|
Returns an iterator over module parameters. |
Registers a backward hook on the module. |
|
|
Adds a buffer to the module. |
Registers a forward hook on the module. |
|
Registers a forward pre-hook on the module. |
|
Registers a backward hook on the module. |
|
|
Adds a parameter to the module. |
|
Change if autograd should record operations on parameters in this module. |
|
This function is called from |
|
Returns a dictionary containing a whole state of the module. |
|
Moves and/or casts the parameters and buffers. |
|
Moves the parameters and buffers to the specified device without copying storage. |
|
Sets the module in training mode. |
|
Casts all parameters and buffers to |
|
Moves all model parameters and buffers to the XPU. |
|
Sets gradients of all model parameters to zero. |