(channels=3, num_classes=10)
| 106 | return ShuffleNet(channels=channels,num_classes=num_classes, dropout=dropout, cfg=cfg) |
| 107 | |
| 108 | def ShuffleNetG3(channels=3, num_classes=10): |
| 109 | cfg = { |
| 110 | 'out_planes': [240,480,960], |
| 111 | 'num_blocks': [4,8,4], |
| 112 | 'groups': 3 |
| 113 | } |
| 114 | return ShuffleNet(channels=channels, num_classes=num_classes, cfg=cfg) |
| 115 | |
| 116 | |
| 117 | def test(): |