| Safe Haskell | None |
|---|---|
| Language | Haskell2010 |
MXNet.NN.Types
- data Parameter a = Parameter {
- _param_in :: NDArray a
- _param_grad :: NDArray a
- data Session a = Session {
- _sess_param :: !(HashMap String (Parameter a))
- _sess_context :: !Context
- sess_param :: forall a a. Lens (Session a) (Session a) (HashMap String (Parameter a)) (HashMap String (Parameter a))
- sess_context :: forall a. Lens' (Session a) Context
- type TrainM a m = StateT (Session a) m
- type Initializer a = Context -> [Int] -> IO (NDArray a)
Documentation
Constructors
| Parameter | |
Fields
| |
Session is all the Parameters and a Context
type Session a = (M.HashMap String (Parameter a), Context)
Constructors
| Session | |
Fields
| |