Skip to content

Commit

Permalink
hubconf documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
chsasank committed May 7, 2020
1 parent beccfb2 commit 0088b7e
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions hubconf.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,9 @@ def _model_fn(num_classes=1000, in_channels=3, pretrained='imagenet'):
Described in detail here: https://arxiv.org/abs/1905.11946
Args:
num_classes (int, optional): Description
in_channels (int, optional): Description
num_classes (int, optional): Number of classes, default is 1000.
in_channels (int, optional): Number of input channels, default
is 3.
pretrained (str, optional): One of [None, 'imagenet', 'advprop']
If None, no pretrained model is loaded.
If 'imagenet', models trained on imagenet dataset are loaded.
Expand Down

0 comments on commit 0088b7e

Please sign in to comment.