--model-mode (str) - which network you use, (example: LARGE, SMALL), (default: LARGE).
--load-pretrained (bool) - (default: False).
--evaluate (bool) - Used when testing. (default: False).
--multiplier (float) - (default: 1.0).
Number of Parameters
import torchfrom model import MobileNetV3def get_model_parameters(model):
total_parameters = 0
for layer in list(model.parameters()):
layer_parameter = 1
for l in list(layer.size()):
layer_parameter *= l
total_parameters += layer_parameter return total_parameters
tmp = torch.randn((128, 3, 224, 224))
model = MobileNetV3(model_mode="LARGE", multiplier=1.0)print("Number of model parameters: ", get_model_parameters(model))