Update qlnet.py
Browse files
qlnet.py
CHANGED
|
@@ -387,21 +387,21 @@ class QLNet(nn.Module):
|
|
| 387 |
return x
|
| 388 |
|
| 389 |
|
| 390 |
-
def _create_qlnet(variant, pretrained=False, **kwargs):
|
| 391 |
-
|
| 392 |
|
| 393 |
|
| 394 |
-
@register_model
|
| 395 |
-
def qlnet22(pretrained=False, **kwargs):
|
| 396 |
-
|
| 397 |
-
|
| 398 |
-
|
| 399 |
-
|
| 400 |
|
| 401 |
|
| 402 |
-
@register_model
|
| 403 |
-
def qlnet26(pretrained=False, **kwargs):
|
| 404 |
-
|
| 405 |
-
|
| 406 |
-
|
| 407 |
-
|
|
|
|
| 387 |
return x
|
| 388 |
|
| 389 |
|
| 390 |
+
# def _create_qlnet(variant, pretrained=False, **kwargs):
|
| 391 |
+
# return build_model_with_cfg(QLNet, variant, pretrained, **kwargs)
|
| 392 |
|
| 393 |
|
| 394 |
+
# @register_model
|
| 395 |
+
# def qlnet22(pretrained=False, **kwargs):
|
| 396 |
+
# """Constructs a QLNet22 model.
|
| 397 |
+
# """
|
| 398 |
+
# model_args = dict(block=QLBlock, layers=[3, 4, 12, 3], **kwargs)
|
| 399 |
+
# return _create_qlnet('qlnet22', pretrained, **dict(model_args, **kwargs))
|
| 400 |
|
| 401 |
|
| 402 |
+
# @register_model
|
| 403 |
+
# def qlnet26(pretrained=False, **kwargs):
|
| 404 |
+
# """Constructs a QLNet26 model.
|
| 405 |
+
# """
|
| 406 |
+
# model_args = dict(block=QLBlock, layers=[3, 8, 12, 3], **kwargs)
|
| 407 |
+
# return _create_qlnet('qlnet26', pretrained, **dict(model_args, **kwargs))
|