Skip to content

Commit

Permalink
added GetModelPath() and GetPrototxtPath()
Browse files Browse the repository at this point in the history
  • Loading branch information
dusty-nv committed Dec 28, 2018
1 parent ec6dae4 commit f0c6264
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions tensorNet.h
Original file line number Diff line number Diff line change
Expand Up @@ -214,6 +214,16 @@ class tensorNet
*/
void SetStream( cudaStream_t stream );

/**
* Retrieve the path to the network prototxt file.
*/
inline const char* GetPrototxtPath() const { return mPrototxtPath.c_str(); }

/**
* Retrieve the path to the network model file.
*/
inline const char* GetModelPath() const { return mModelPath.c_str(); }

protected:

/**
Expand Down

0 comments on commit f0c6264

Please sign in to comment.