modelsummary
modelsummary copied to clipboard
All Model summary in PyTorch similar to `model.summary()` in Keras
Results
3
modelsummary issues
Sort by
recently updated
recently updated
newest added
When a model takes multiple inputs (ex. a Siamese network) that all go through the same set of layers before the results are compared later on in the network, the...
On running the following code I get the error above ```python def get_model(**kwargs): model = models.densenet121(pretrained=True).to(device) # Freeze parameters so we don't backprop through them for param in model.parameters(): param.requires_grad...