Skip to content

Conversation

lopuhin
Copy link
Contributor

@lopuhin lopuhin commented Jul 14, 2017

densenet121 and other functions in densenet module accept **kwargs, but they are not passed to DenseNet class, unlike for other models (for example vgg and resnet). They are useful to define e.g. num_classes different from default ImageNet value.

@soumith soumith merged commit 47bc815 into pytorch:master Jul 14, 2017
@soumith
Copy link
Member

soumith commented Jul 14, 2017

thanks!

@lopuhin lopuhin deleted the densenet-kwargs branch July 14, 2017 14:26
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants