public class NetworkMemoryReport extends MemoryReport
MultiLayerNetwork
or ComputationGraph
,
composed of multiple layers and/or vertices.CACHE_MODE_ALL_ZEROS
Constructor and Description |
---|
NetworkMemoryReport(@NonNull Map<String,MemoryReport> layerAndVertexReports,
@NonNull Class<?> modelClass,
String modelName,
InputType... networkInputTypes) |
Modifier and Type | Method and Description |
---|---|
long |
getMemoryBytes(MemoryType memoryType,
int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode,
DataType dataType)
Get the memory estimate (in bytes) for the specified type of memory
|
String |
getName()
Name of the object that the memory report was generated for
|
Class<?> |
getReportClass() |
long |
getTotalMemoryBytes(int minibatchSize,
@NonNull MemoryUseMode memoryUseMode,
@NonNull CacheMode cacheMode,
@NonNull DataType dataType)
Get the total memory use in bytes for the given configuration
|
String |
toString() |
cacheModeMapFor, fromJson, fromYaml, getBytesPerElement, getMemoryBytes, getTotalMemoryBytes, toJson, toYaml
public Class<?> getReportClass()
getReportClass
in class MemoryReport
public String getName()
MemoryReport
getName
in class MemoryReport
public long getTotalMemoryBytes(int minibatchSize, @NonNull @NonNull MemoryUseMode memoryUseMode, @NonNull @NonNull CacheMode cacheMode, @NonNull @NonNull DataType dataType)
MemoryReport
getTotalMemoryBytes
in class MemoryReport
minibatchSize
- Mini batch size to estimate the memory formemoryUseMode
- The memory use mode (training or inference)cacheMode
- The CacheMode to usedataType
- Nd4j datatypepublic long getMemoryBytes(MemoryType memoryType, int minibatchSize, MemoryUseMode memoryUseMode, CacheMode cacheMode, DataType dataType)
MemoryReport
getMemoryBytes
in class MemoryReport
memoryType
- Type of memory to get the estimate for invitesminibatchSize
- Mini batch size to estimate the memory formemoryUseMode
- The memory use mode (training or inference)cacheMode
- The CacheMode to usedataType
- Nd4j datatypepublic String toString()
toString
in class MemoryReport
Copyright © 2020. All rights reserved.