#Flops

cszn

KAIR

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

vra

flopth

A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.

gmalivenko

onnx-opcounter

Count number of parameters / MACs / FLOPS for ONNX models.

kyo-takano

chinchilla

A toolkit for scaling law research ⚖

RobinKa

discrete-flow-matching-pytorch

Discrete Flow Matching implemented in PyTorch