7 lines
138 B
Python
7 lines
138 B
Python
|
from ml_pipeline import config
|
||
|
|
from ml_pipeline.model.cnn import VGG11
|
||
|
|
|
||
|
|
def test_in_channels():
|
||
|
|
assert config.model.name == 'vgg11'
|
||
|
|
|