Image Classification
timm
PDE
ConvNet
liuyao commited on
Commit
2b29261
1 Parent(s): d7157c8

Update qlnet.py

Browse files
Files changed (1) hide show
  1. qlnet.py +14 -14
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
- 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))
 
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))