diff --git a/CNNDemo/Resnet50.py b/CNNDemo/Resnet50.py index e44a955..8743007 100644 --- a/CNNDemo/Resnet50.py +++ b/CNNDemo/Resnet50.py @@ -300,26 +300,8 @@ for batch_idx, (data, target) in enumerate(val_loader): currentbyte,binaryfile,strg = genData("verify_input", data, currentbyte, binaryfile, strg) x = resnet50.conv1(data) currentbyte,binaryfile,strg = genData("verify_conv1", x, currentbyte, binaryfile, strg) - - - print("bias") - print(resnet50.bn1.bias.data) - print("weight") - print(resnet50.bn1.weight.data) - print("mean") - print(resnet50.bn1.running_mean.data) - print("var") - print(resnet50.bn1.running_var.data) - print("input") - print(x) - x = resnet50.bn1(x) currentbyte,binaryfile,strg = genData("verify_bn1", x, currentbyte, binaryfile, strg) - - - print("output") - print(x) - x = resnet50.relu(x) currentbyte,binaryfile,strg = genData("verify_relu", x, currentbyte, binaryfile, strg) x = resnet50.maxpool(x)