projects / platform / core / ml / nntrainer.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 7fecf18) | patch
[layers] Support for weight decay to layers
authorParichay Kapoor <pk.kapoor@samsung.com>
Mon, 24 Jan 2022 09:54:32 +0000 (18:54 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Wed, 9 Feb 2022 09:34:12 +0000 (18:34 +0900)
commit62ab2b4e3adc8714482afec7c7fa191ce63b5455
tree8f15647da69728d834e69fd6246e500b7cbc069ftree | snapshot
parent7fecf186ffe2e6841744cd03a632e7c8a52e360fcommit | diff
[layers] Support for weight decay to layers

Add support for weight decay to the layers with weights.
Further update the requestWeight API of the layer context to accept
weight decay, and update the usage at manager.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
19 files changed:
nntrainer/graph/network_graph.cpp diff | blob | history
nntrainer/layers/bn_layer.cpp diff | blob | history
nntrainer/layers/centroid_knn.cpp diff | blob | history
nntrainer/layers/conv2d_layer.cpp diff | blob | history
nntrainer/layers/embedding.cpp diff | blob | history
nntrainer/layers/fc_layer.cpp diff | blob | history
nntrainer/layers/gru.cpp diff | blob | history
nntrainer/layers/grucell.cpp diff | blob | history
nntrainer/layers/layer_context.h diff | blob | history
nntrainer/layers/lstm.cpp diff | blob | history
nntrainer/layers/lstmcell.cpp diff | blob | history
nntrainer/layers/mol_attention_layer.cpp diff | blob | history
nntrainer/layers/rnn.cpp diff | blob | history
nntrainer/layers/rnncell.cpp diff | blob | history
nntrainer/layers/zoneout_lstmcell.cpp diff | blob | history
nntrainer/tensor/manager.cpp diff | blob | history
nntrainer/tensor/tensor_wrap_specs.h diff | blob | history
nntrainer/tensor/weight.cpp diff | blob | history
nntrainer/tensor/weight.h diff | blob | history
Domain: Machine Learning / ML Framework;