Merge branch 'master' of https://github.com/wang-xinyu/tensorrtx
This commit is contained in:
commit
9b60aa240f
@ -4,7 +4,7 @@
|
||||
// Load weights from files.
|
||||
// TensorRT weight files have a simple space delimited format:
|
||||
// [type] [size] <data x size in hex>
|
||||
std::map<std::string, Weights> loadWeights(const std::string input) {
|
||||
std::map<std::string, Weights> loadWeights(const std::string file) {
|
||||
std::cout << "Loading weights: " << file << std::endl;
|
||||
std::map<std::string, Weights> weightMap;
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user