nntrainer icon indicating copy to clipboard operation
nntrainer copied to clipboard

[Layer] Add checkpoint property

Open jihochu opened this issue 1 year ago • 2 comments

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]

jihochu avatar Apr 27 '23 02:04 jihochu