@KFrank



 Capo_Mestre:

Only when the function returns the result transfered on cpu, detached and converted to numpy, the minimize function can accept MSE_toOptimize as its first argument. This probably breaks the flow of computation graph, but I am not sure how to make minimize work otherwise. Would you be aware of a simple workaround solution?


I found a code here 1 that sounds like a workaround solution, but for now it is not quite transparent how to use it in my situation.