Skip to content

GitLab

"examples/dnn_introduction2_ex.cpp" did not exist on "1f0705ae92d95b085bf9487770e8287c4c07fc79"
Commit a11cb507 authored by Xiaoliang Dai's avatar Xiaoliang Dai Committed by Facebook GitHub Bot
Browse files

Support customized subclass selection 

Summary: Support customized subclass selection.  Only the selected gestures are used for model training.

Reviewed By: sanjeevk42

Differential Revision: D30205443

fbshipit-source-id: 36337893aa5d06bb8be5d5587da11398b246b02e
parent 5e521841
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
d2go/modeling/subclass.py
View file @ a11cb507
...@@ -72,7 +72,7 @@ class SubclassDatasetMapper(D2GoDatasetMapper): ...@@ -72,7 +72,7 @@ class SubclassDatasetMapper(D2GoDatasetMapper):
super().__init__(cfg, is_train=is_train, tfm_gens=tfm_gens) super().__init__(cfg, is_train=is_train, tfm_gens=tfm_gens)
if subclass_fetcher is None: if subclass_fetcher is None:
fetcher_name = cfg.MODEL.SUBCLASS.SUBCLASS_ID_FETCHER fetcher_name = cfg.MODEL.SUBCLASS.SUBCLASS_ID_FETCHER
self.subclass_fetcher = SUBCLASS_FETCHER_REGISTRY.get(fetcher_name)() self.subclass_fetcher = SUBCLASS_FETCHER_REGISTRY.get(fetcher_name)(cfg)
logger.info( logger.info(
f"Initialized {self.__class__.__name__} with " f"Initialized {self.__class__.__name__} with "
f"subclass fetcher '{self.subclass_fetcher.__class__.__name__}'" f"subclass fetcher '{self.subclass_fetcher.__class__.__name__}'"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment