Commit 8d072ebf authored by Yanghan Wang's avatar Yanghan Wang Committed by Facebook GitHub Bot
Browse files

add print during _populate_registries

Summary: Pull Request resolved: https://github.com/facebookresearch/d2go/pull/634

Reviewed By: yzhao30

Differential Revision: D51208655

fbshipit-source-id: 3280bde8807b623ec56841cc6d0ffc87a1e02e83
parent 40e78153
#!/usr/bin/env python3
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
import logging
from d2go.registry.bootstrap import bootstrap_registries
from mobile_cv.common.misc.oss_utils import fb_overwritable
logger = logging.getLogger(__name__)
_INITIALIZED = False
......@@ -46,9 +50,11 @@ def _register_builtin_datasets():
@fb_overwritable()
def _populate_registries():
logger.info("Populating registries ...")
from d2go import optimizer # noqa
from d2go.data import dataset_mappers # noqa
from d2go.modeling.backbone import fbnet_v2 # noqa
logger.info(f"populated registries for following modules: {optimizer} {dataset_mappers} {fbnet_v2}")
# fmt: on
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