Lightning Model module , get_model_size method
🚀 Feature
A get_model_size method in model module which return model size in megabyte based on precision.
Motivation
Just thinking out loud..
Additional context
We can add this in mode summary too
# ------------
# model
# ------------
model = LitClassifier(args.hidden_dim, args.learning_rate)
>> print(model.size)
>> 104.14281 Mb with fp32.