deviceMemorySize
The device memory size in bytes for all intermediate tensors for forward, gradient passes, and optimizer updates for all layers in the training graph.
Declaration
var deviceMemorySize: Int { get }The device memory size in bytes for all intermediate tensors for forward, gradient passes, and optimizer updates for all layers in the training graph.
var deviceMemorySize: Int { get }