JiaquanYe / TableMASTER-mmocr

2nd solution of ICDAR 2021 Competition on Scientific Literature Parsing, Task B.
Apache License 2.0
410 stars 100 forks source link

请问TableMaster可以用MMdet的get_flops算参数量和GFLOPs吗?NotImplementedError: FLOPs counter is currently not currently supported with TABLEMASTER #75

Open MathamPollard opened 2 months ago

MathamPollard commented 2 months ago

报错: NotImplementedError: FLOPs counter is currently not currently supported with TABLEMASTER 报错片段: model = build_detector( cfg.model, train_cfg=cfg.get('train_cfg'), test_cfg=cfg.get('test_cfg')) if torch.cuda.is_available(): model.cuda() model.eval() print(model) if hasattr(model, 'forward_dummy'): model.forward = model.forward_dummy else: raise NotImplementedError( 'FLOPs counter is currently not currently supported with {}'. format(model.class.name))