Skip to content

tenzr::optim::OptState

Optimizer state holding the step counter and per-parameter tensors.

#include <transform.h>

Public Attributes

Name
uint64_t step
std::unordered_map< std::string, std::unordered_map< std::string, Tensor > > param_states

Public Attributes Documentation

variable step

uint64_t step {0};

variable param_states

std::unordered_map< std::string, std::unordered_map< std::string, Tensor > > param_states;