mindscience.common
The Adahessian optimizer, which performs optimization using second-order information from the diagonal elements of the Hessian matrix. |
|
Calculate Hessian matrix of network model. |
|
Calculate Jacobian matrix of network model. |
|
Construct 2D sine-cosine positional embeddings on a two-dimensional grid. |
|
Gets the loss function. |
|
Generate decay learning rate array of each parameter group by gamma once the number of epoch reaches one of the milestones. |
|
Generate polynomial decay learning rate array. |
|
Calculates learning rate base on cosine decay function. |
|
Glorot uniform. |
|
Lecun init. |
|
Compute the MTL strategy weighted multi-task losses automatically. |
|
Convert an image-like tensor into a sequence of flattened patches. |
|
Applies a pixel_shuffle operation over an input signal composed of several input planes. |
|
Applies a pixel_unshuffle operation over an input signal composed of several input planes. |
|
Applies a pixelshuffle operation over an input signal composed of several input planes. |
|
Applies a pixelunshuffle operation over an input signal composed of several input planes. |
|
Relative Root Mean Square Error (RRMSE) is the root mean squared error normalized by the root-mean-square value where each residual is scaled against the actual value. |
|
Applies spectral normalization to a parameter in the given module. |
|
Convert an integer or a tuple of integers to a 2-tuple. |
|
Convert an integer or a tuple of integers to a 3-tuple. |
|
Convert a sequence of flattened patches back into an image-like tensor. |
|
The multi-level wavelet transformation losses. |