diff --git a/demo.py b/demo.py index ebc5012..cf93edf 100644 --- a/demo.py +++ b/demo.py @@ -54,6 +54,7 @@ cfg = Config(args) model_config = cfg.model_cfg model_config.device_8bit = args.gpu_id model_cls = registry.get_model_class(model_config.arch) +print(model_config) model = model_cls.from_config(model_config).to('cuda:{}'.format(args.gpu_id)) # TODO: Fix hard-coding `cc12m`