mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
Start implementing Inception v3
This commit is contained in:
17
examples/inception_v3/inception_v3.py
Normal file
17
examples/inception_v3/inception_v3.py
Normal file
@@ -0,0 +1,17 @@
|
||||
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()
|
||||
|
||||
summary(inception, (3, 299, 299))
|
||||
Reference in New Issue
Block a user