def load_model(args): model = test_engine.initialize_model_from_cfg(args.wts_file) blobs = mutils.get_ws_blobs() return model, blobs
def load_model(args): model = test_engine.initialize_model_from_cfg(cfg.TEST.WEIGHTS) blobs = mutils.get_ws_blobs() return model, blobs