This commit is contained in:
wang-xinyu 2021-04-27 08:39:48 +00:00
commit 9b60aa240f

View File

@ -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;