how did i miss example_inputs

README.md

@@ -48,6 +48,10 @@ url = f"https://huggingface.co/FredZhang7/efficientnetv2.5_rw_s/resolve/main/{mo
 file_name = f"./{base_model}.pth"
 urllib.request.urlretrieve(url, file_name)
 
+shape = (2,) + input_size
+example_inputs = torch.randn(shape)
+example_inputs = (example_inputs - example_inputs.min()) / (example_inputs.max() - example_inputs.min())
+
 model = torch.load(file_name)
 model.classifier = torch.nn.Linear(in_features=1984, out_features=nclass, bias=True)
 macs, nparams = get_model_complexity_info(model, input_size, as_strings=False, print_per_layer_stat=print_layer_stats, verbose=verbose)