From da95fa203e2f3940a1ebf40f70957f5d58fb1802 Mon Sep 17 00:00:00 2001 From: LordMathis Date: Thu, 30 May 2024 13:19:11 +0200 Subject: [PATCH] Update inception py --- examples/inception_v3/inception_v3.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/examples/inception_v3/inception_v3.py b/examples/inception_v3/inception_v3.py index 0570af7..869ee04 100644 --- a/examples/inception_v3/inception_v3.py +++ b/examples/inception_v3/inception_v3.py @@ -1,17 +1,14 @@ -import torch import torchvision import sys -from torchsummary import summary - -inception = torchvision.models.inception_v3(weights=torchvision.models.Inception_V3_Weights.DEFAULT) -inception.eval() - sys.path.append('../../tools') # Ugly hack from utils import export_model_weights, print_model_parameters -print_model_parameters(inception) # print layer names and number of parameters -inception.cuda() +if __name__ == "__main__": + inception = torchvision.models.inception_v3(weights=torchvision.models.Inception_V3_Weights.DEFAULT) + inception.eval() -summary(inception, (3, 299, 299)) \ No newline at end of file + print_model_parameters(inception) # print layer names and number of parameters + + export_model_weights(inception, 'inception_v3_weights.bin') \ No newline at end of file