Skip to content

Commit bafe9c8

Browse files
committed
update basic pytorch simplenet.py
1 parent bbdbc20 commit bafe9c8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ImageNet/simplenet.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -257,12 +257,12 @@ def __init__(
257257
self.in_chans = in_chans
258258
self.scale = scale
259259
self.networks = [
260-
"simplenetv1_imagenet", # 0
260+
"simplenetv1_imagenet", # 0
261261
"simplenetv1_imagenet_9m", # 1
262262
# other archs
263-
"simplenet_cifar_310k", # 2
264-
"simplenet_cifar_460k", # 3
265-
"simplenet_cifar_5m", # 4
263+
"simplenet_cifar_310k", # 2
264+
"simplenet_cifar_460k", # 3
265+
"simplenet_cifar_5m", # 4
266266
"simplenet_cifar_5m_extra_pool", # 5
267267
]
268268
self.network_idx = network_idx

0 commit comments

Comments
 (0)