mirror of https://github.com/hpcaitech/ColossalAI
aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
617 B
17 lines
617 B
#!/usr/bin/env python |
|
# -*- encoding: utf-8 -*- |
|
|
|
from pathlib import Path |
|
|
|
from colossalai.context.config import Config |
|
|
|
|
|
def test_load_config(): |
|
filename = Path(__file__).parent.joinpath("sample_config.py") |
|
config = Config.from_file(filename) |
|
|
|
assert config.train_data, "cannot access train data as attribute" |
|
assert config.train_data.dataset, "cannot access grandchild attribute" |
|
assert isinstance( |
|
config.train_data.dataset.transform_pipeline[0], dict |
|
), f"expected attribute transform_pipeline elements to be a dict, but found {type(config.train_data.dataset.transform_pipeline)}"
|
|
|