ThanatosShinji / onnx-tool

A parser, editor and profiler tool for ONNX models.
https://pypi.org/project/onnx-tool/
MIT License
399 stars 52 forks source link

request VAE encoder FLOPS #82

Open handsomeZhuang opened 5 months ago

handsomeZhuang commented 5 months ago

hello, could you tell me where the VAE encoder and decoder MACs is?

ThanatosShinji commented 5 months ago

here: https://github.com/ThanatosShinji/onnx-tool?tab=readme-ov-file#results-of-onnx-model-zoo-and-sota-models

handsomeZhuang commented 5 months ago

here: https://github.com/ThanatosShinji/onnx-tool?tab=readme-ov-file#results-of-onnx-model-zoo-and-sota-models image do it refers to [encoder]566,371MACS/[decoder]1,271,959MACS every image?