The memory transfer for gradient computation and back propagation have not been done will need to resolve that.
The memory transfer for gradient computation and back propagation have not been done will need to resolve that.