Open aninda-leonardo opened 9 months ago
We often need to supply keyword arguments for model forward pass.
It's probably also worth returning the model outputs along the flops and param counts, but I did not want to change the output signature in this one.
We often need to supply keyword arguments for model forward pass.
It's probably also worth returning the model outputs along the flops and param counts, but I did not want to change the output signature in this one.