mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
add comment and static decorator (#60)
This commit is contained in:
parent
d157243077
commit
fbce75e6bc
@ -105,6 +105,9 @@ class MMCodebase(metaclass=ABCMeta):
|
||||
pass
|
||||
|
||||
|
||||
# Note that the build function returns the class instead of its instance.
|
||||
|
||||
|
||||
def __build_codebase_class(codebase: Codebase, registry: Registry):
|
||||
return registry.module_dict[codebase.value]
|
||||
|
||||
|
@ -71,6 +71,7 @@ class MMClassification(MMCodebase):
|
||||
|
||||
return dataset
|
||||
|
||||
@staticmethod
|
||||
def build_dataloader(dataset: Dataset,
|
||||
samples_per_gpu: int,
|
||||
workers_per_gpu: int,
|
||||
|
Loading…
x
Reference in New Issue
Block a user