mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

ResNet working

Browse Source
This commit is contained in:
Davide Sapienza
2019-10-29 17:21:31 +01:00
parent e9ec582223
commit 02936fa928
5 changed files with 242 additions and 317 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/LayerWgs.cpp
-18
View File
@@ -20,13 +20,6 @@ LayerWgs::LayerWgs(Network *net, int inputs, int outputs,
seek += inputs*outputs*kh*kw*kl;
readBinaryFile(weights_path.c_str(), outputs, &bias_h, &bias_d, seek);
std::cout<<"w:\n";
printDeviceVector(64, data_d, true);
std::cout<<"b:\n";
printDeviceVector(64, bias_d, true);
std::cout<<"step----------------------------------------\n";
this->batchnorm = batchnorm;
if(batchnorm) {
seek += outputs;
@@ -36,17 +29,6 @@ std::cout<<"step----------------------------------------\n";
seek += outputs;
readBinaryFile(weights_path.c_str(), outputs, &variance_h, &variance_d, seek);
std::cout<<"s:\n";
printDeviceVector(64, scales_d, true);
std::cout<<"m:\n";
printDeviceVector(64, mean_d, true);
std::cout<<"v:\n";
printDeviceVector(64, variance_d, true);
std::cout<<"END----------------------------------------\n";
float eps = CUDNN_BN_MIN_EPSILON;
power_h = new dnnType[outputs];