Accurate method to determine the model size

[mustansarsaeed]

Hi, I am using syft 0.2.x. I want to determine the size of the model. What is the correct way to determine the size? I have two options:

Option: 1

val modelFile = File(modelFileLocation)
modelFile.length()

Option 2

SyftState.loadSyftState(modelFile)
size of syftstate

In terms of model size and number of parameters, SyftState should be used or modelFile.length()?

Thanks.

[vkkhare]

@mustansarsaeed number of parameters can be calculated from model

val num_params = model.getParamArray().reduce{ sum, tensor ->
    sum + tensor.shape().reduce{ m,e -> m*e }
}