Update qlnet.py
Browse files
qlnet.py
CHANGED
@@ -262,8 +262,8 @@ class QLNet(nn.Module):
|
|
262 |
self.num_classes = num_classes
|
263 |
self.grad_checkpointing = False
|
264 |
|
265 |
-
act_layer = get_act_layer(act_layer)
|
266 |
-
norm_layer = get_norm_layer(norm_layer)
|
267 |
|
268 |
# Stem
|
269 |
deep_stem = 'deep' in stem_type
|
|
|
262 |
self.num_classes = num_classes
|
263 |
self.grad_checkpointing = False
|
264 |
|
265 |
+
# act_layer = get_act_layer(act_layer)
|
266 |
+
# norm_layer = get_norm_layer(norm_layer)
|
267 |
|
268 |
# Stem
|
269 |
deep_stem = 'deep' in stem_type
|