|
|
@@ -15,7 +15,7 @@ |
|
|
"""hub config.""" |
|
|
"""hub config.""" |
|
|
from src.resnet import resnet50, resnet101, se_resnet50 |
|
|
from src.resnet import resnet50, resnet101, se_resnet50 |
|
|
|
|
|
|
|
|
def create_network(name, **kwargs): |
|
|
|
|
|
|
|
|
def create_network(name, *args, **kwargs): |
|
|
if name == 'resnet50': |
|
|
if name == 'resnet50': |
|
|
return resnet50(*args, **kwargs) |
|
|
return resnet50(*args, **kwargs) |
|
|
if name == 'resnet101': |
|
|
if name == 'resnet101': |
|
|
|