hubconf.py 547 字节
Newer Older
L
for hub  
lyuwenyu 已提交
1

L
lyuwenyu 已提交
2
dependencies = ['paddle', 'numpy']
L
for hub  
lyuwenyu 已提交
3 4


L
lyuwenyu 已提交
5 6 7
from ppcls.modeling.architectures.resnet import ResNet18 as _ResNet18
from ppcls.modeling.architectures.resnet import ResNet34 as _ResNet34
from ppcls.modeling.architectures.resnet import ResNet50 as _ResNet50
L
for hub  
lyuwenyu 已提交
8 9


L
lyuwenyu 已提交
10 11 12

def ResNet18(**kwargs):
    '''ResNet18
L
for hub  
lyuwenyu 已提交
13
    '''
L
lyuwenyu 已提交
14
    model = _ResNet18(**kwargs)
L
for hub  
lyuwenyu 已提交
15 16
    return model

L
lyuwenyu 已提交
17 18 19 20 21


def ResNet34(**kwargs):
    '''ResNet34
    '''
L
update  
lyuwenyu 已提交
22
    model = _ResNet34(**kwargs)
L
lyuwenyu 已提交
23 24 25 26 27 28 29
    return model



def ResNet50(**kwargs):
    '''ResNet50
    '''
L
update  
lyuwenyu 已提交
30
    model = _ResNet50(**kwargs)
L
lyuwenyu 已提交
31
    return model