lazyslide_models.base.ModelBase

class ModelBase

Bases: ABC

estimate_flops(*args, method='forward', **kwargs)

Count the number of flops in a model.

estimate_param_size()

Count the number of parameters in a model.