![]() |
MyCaffe
1.12.2.41
Deep learning software for Windows C# programmers.
|
The GPUParams contains the connection to the low-level Cuda, and the stream associated with this instance. More...
Public Member Functions | |
| GPUParams (CudaDnn< T > cuda, Log log, Solver< T > root_solver, int nDeviceID) | |
| The GPUParams constructor. More... | |
| void | Dispose () |
| Release all GPU and Host resources used. More... | |
| void | SynchronizeStream () |
| Synchronize with the Cuda stream. More... | |
| void | Configure (Solver< T > solver) |
| Configure the GPU Params by copying the Solver training Net parameters into the data and diff buffers. More... | |
| void | apply_buffers (BlobCollection< T > rgBlobs, long hBuffer, long lTotalSize, Op op) |
| Transfer between the data/diff buffers and a collection of Blobs (e.g. the learnable parameters). More... | |
Public Member Functions inherited from MyCaffe.common.Params< T > | |
| Params (Solver< T > root_solver) | |
| The Param constructor. More... | |
Protected Attributes | |
| CudaDnn< T > | m_cuda |
| The instance of CudaDnn that provides the connection to Cuda. More... | |
| Log | m_log |
| The Log used for output. More... | |
| long | m_hStream |
| The handle to the Cuda stream used for synchronization. More... | |
Protected Attributes inherited from MyCaffe.common.Params< T > | |
| long | m_lCount |
| size of the buffers (in items). More... | |
| long | m_lExtra |
| size of the padding added to the memory buffers. More... | |
| long | m_hData |
| Handle to GPU memory containing the Net parameters. More... | |
| long | m_hDiff |
| Handle to GPU memory containing the Net gradient. More... | |
| int | m_nDeviceID |
| The Device ID. More... | |
Additional Inherited Members | |
Properties inherited from MyCaffe.common.Params< T > | |
| long | count [get] |
| Returns the size of the buffers (in items). More... | |
| long | data [get] |
| Returns the handle to the GPU memory containing the Net parameters. More... | |
| long | diff [get] |
| Returns the handle to the GPU memory containing the Net gradients. More... | |
The GPUParams contains the connection to the low-level Cuda, and the stream associated with this instance.
| T | Specifies the base type float or double. Using float is recommended to conserve GPU memory. |
Definition at line 103 of file Parallel.cs.
| MyCaffe.common.GPUParams< T >.GPUParams | ( | CudaDnn< T > | cuda, |
| Log | log, | ||
| Solver< T > | root_solver, | ||
| int | nDeviceID | ||
| ) |
The GPUParams constructor.
| cuda | Specifies the CudaDnn connection to Cuda. |
| log | Specifies the Log for output. |
| root_solver | Specifies the root Solver. |
| nDeviceID | Specifies the device ID to use for this instance. |
Definition at line 144 of file Parallel.cs.
| void MyCaffe.common.GPUParams< T >.apply_buffers | ( | BlobCollection< T > | rgBlobs, |
| long | hBuffer, | ||
| long | lTotalSize, | ||
| Op | op | ||
| ) |
Transfer between the data/diff buffers and a collection of Blobs (e.g. the learnable parameters).
| rgBlobs | Specifies the collection of Blobs to transfer data with. |
| hBuffer | Specifies a handle to the memory on the GPU to transfer with the Blob collection. |
| lTotalSize | Specifies the number of items to transfer. |
| op | Specifies the type of transfer to perform. |
Definition at line 221 of file Parallel.cs.
| void MyCaffe.common.GPUParams< T >.Configure | ( | Solver< T > | solver | ) |
Configure the GPU Params by copying the Solver training Net parameters into the data and diff buffers.
| solver |
Definition at line 207 of file Parallel.cs.
| void MyCaffe.common.GPUParams< T >.Dispose | ( | ) |
Release all GPU and Host resources used.
Definition at line 174 of file Parallel.cs.
| void MyCaffe.common.GPUParams< T >.SynchronizeStream | ( | ) |
Synchronize with the Cuda stream.
Definition at line 198 of file Parallel.cs.
|
protected |
The instance of CudaDnn that provides the connection to Cuda.
Definition at line 108 of file Parallel.cs.
|
protected |
The handle to the Cuda stream used for synchronization.
Definition at line 116 of file Parallel.cs.
|
protected |
The Log used for output.
Definition at line 112 of file Parallel.cs.