Package | Description |
---|---|
org.nd4j.autodiff.samediff.internal |
Modifier and Type | Class and Description |
---|---|
class |
InferenceSession
InferenceSession: Performs inference (forward pass) on a SameDiff instance to get the outputs of the requested nodes.
Dynamically (in AbstractSession) calculates the required subgraph to execute to get the required outputs. Note that while AbstractSession handles the graph structure component, InferenceSession handles only op execution and memory management For INDArray memory management - i.e., tracking and releasing memory manually, as soon as possible, to minimize memory use - this is implemented using a SessionMemMgr instance (for allocations/deallocations) and
also IdentityDependencyTracker to track where arrays are actually used. |
class |
TrainingSession
TrainingSession extends InferenceSession, to add training-specific functionality:
- Application of regularization (L1, L2, weight decay etc) - Inline updating of variables, using updater/optimizer (Adam, Nesterov, SGD, etc) - Calculation of regularization scores (Score for L1, L2, etc) |
Copyright © 2019. All rights reserved.