mitudesk/uma_diffsvc

1import os2 3os.environ["OMP_NUM_THREADS"] = "1"4 5import importlib6from utils.hparams import set_hparams, hparams7 8 9def binarize():10    binarizer_cls = hparams.get("binarizer_cls", 'basics.base_binarizer.BaseBinarizer')11    pkg = ".".join(binarizer_cls.split(".")[:-1])12    cls_name = binarizer_cls.split(".")[-1]13    binarizer_cls = getattr(importlib.import_module(pkg), cls_name)14    print("| Binarizer: ", binarizer_cls)15    binarizer_cls().process()16 17 18if __name__ == '__main__':19    set_hparams()20    binarize()21