nntrainer
nntrainer copied to clipboard
[Layer] Add checkpoint property
This patch adds CheckPoint property to layer node. Default value is checkpointed, which can store the layer data for the backward step. laoded and unloaded state are for unchecked nodes. The data would be unloaded from memory, and reloaded when it's needed when the backward step.
Signed-off-by: Jiho Chu [email protected]