github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 12:01:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typo tests/ (#3936)

Browse Source
This commit is contained in:
digger yu
2023-06-09 09:49:41 +08:00
committed by GitHub
parent bd2c7c3297
commit e61ffc77c6
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/kit/model_zoo/registry.py
View File

@@ -2,7 +2,7 @@
from dataclasses import dataclass
from typing import Callable
__all__ = ['ModelZooRegistry', 'ModelAttributem', 'model_zoo']
__all__ = ['ModelZooRegistry', 'ModelAttribute', 'model_zoo']
@dataclass
@@ -37,7 +37,7 @@ class ModelZooRegistry(dict):
>>> model_zoo = ModelZooRegistry()
>>> model_zoo.register('resnet18', resnet18, resnet18_data_gen)
>>> # Run the model
>>> data = resnresnet18_data_gen() # do not input any argument
>>> data = resnet18_data_gen() # do not input any argument
>>> model = resnet18() # do not input any argument
>>> out = model(**data)