Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Inception B test

Browse Source
This commit is contained in:
Matúš Námešný
2024-06-04 22:26:30 +02:00
parent 99125a0d38
commit 2dcaba40d6
2 changed files with 11922 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/inception_v3/tests/inception_b.py
View File

@@ -34,6 +34,10 @@ with torch.no_grad():
print_cpp_vector(torch.flatten(inception_b.branch3x3dbl_2.bn.weight), "branch3x3dbl_2_bn_weights")
print_cpp_vector(torch.flatten(inception_b.branch3x3dbl_2.bn.bias), "branch3x3dbl_2_bn_bias")
print_cpp_vector(torch.flatten(inception_b.branch3x3dbl_3.conv.weight), "branch3x3dbl_3_conv_weights")
print_cpp_vector(torch.flatten(inception_b.branch3x3dbl_3.bn.weight), "branch3x3dbl_3_bn_weights")
print_cpp_vector(torch.flatten(inception_b.branch3x3dbl_3.bn.bias), "branch3x3dbl_3_bn_bias")
input_shape = (1, 3, 8, 8)
input = torch.randn(input_shape)
print_cpp_vector(torch.flatten(input), "input")

11895
examples/inception_v3/tests/test_inception_b.cpp
View File

File diff suppressed because it is too large Load Diff