Inception v3 - Githubissues

PPPW / deep-learning-random-explore

194 stars 34 forks source link

Inception v3 #5

Closed austinmw closed 5 years ago

austinmw commented 5 years ago

Hi, were you able to get a Learner to work for either torchvision inception_v3 or cadene inceptionv3? I'm able to use Learner directly rather than create_cnn, but learn.summary() fails with AttributeError: 'NoneType' object has no attribute 'shape'

PPPW commented 5 years ago

Hi @austinmw, I have added a hacky example to use torchvision inception v3 for your reference. I'm not sure about your error, need more info from the stack trace.

austinmw commented 5 years ago

@PPPW Thanks! That would have probably taken me quite a while to figure out.

Edit: Are you able to print learn.summary() for the inception model? I can create the learn object but when I try to print summary I get the error AttributeError: 'NoneType' object has no attribute 'shape'.