unfortunately im still having problem to implement that
(not sure what initialization i need to use and how to use the init class)
is it possible some how to “save” the weights before the cross validation and
set the “saved” weights after each itiration ?