Skip to content

Commit

Permalink
Added squeeze_net.py. Thank Yu Gu for converting.
Browse files Browse the repository at this point in the history
  • Loading branch information
yuanming-hu committed Apr 26, 2018
1 parent 4dfada2 commit 9336ce3
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions squeeze_net.py
Expand Up @@ -6,7 +6,6 @@
POOLING_LAYERS = [1, 3, 5]
MODEL_PATH = 'data/squeeze_net/model.pkl'


class SqueezeNet(object):

def __init__(self, imgs):
Expand Down Expand Up @@ -234,4 +233,3 @@ def create_convnet(imgs):
sn = SqueezeNet(imgs)
return {'features_out': sn.get_features_out()}


0 comments on commit 9336ce3

Please sign in to comment.