flopth
flopth copied to clipboard
vra
A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.
The GCT pytorch op is not supported, even though it is no an official layer but a smaller sub-network taken from the [repo](https://github.com/z-x-yang/GCT/blob/master/PyTorch/GCT.py). Any tips on how to incorporate this...
AttributeError Traceback (most recent call last) [](https://localhost:8080/#) in () 1 dummy_inputs = torch.rand(1, 3, 224, 224) ----> 2 flops, params = flopth(matcher, inputs=({"image0": feats0, "image1": feats1})) 3 print(flops, params) 1...
