Open jgbradley1 opened 5 years ago
What input size do the pretrained models expect? I am trying to print a model summary with this approach using torchsummary.
What input size do the pretrained models expect? I am trying to print a model summary with this approach using torchsummary.