sure we can do that. I guess the main thing that is unavoidable is that we need to bring the array from GPU to CPU at each epoch, I guess its unavoidable…unless I create a GPU array that I add the value at each epoch.