From 96ccde9d7a6cf58c26fd224020d4face0f86ee14 Mon Sep 17 00:00:00 2001
From: Aditya Oke <47158509+oke-aditya@users.noreply.github.com>
Date: Tue, 22 Sep 2020 19:15:01 +0530
Subject: [PATCH] Small documentation fix (#3589)

* fixes resnet

* Fixes formatting and resnet
---
 docs/source/transfer_learning.rst | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/docs/source/transfer_learning.rst b/docs/source/transfer_learning.rst
index 4cc1c82335..ba44203721 100644
--- a/docs/source/transfer_learning.rst
+++ b/docs/source/transfer_learning.rst
@@ -54,9 +54,7 @@ Example: Imagenet (computer Vision)
         def __init__(self):
             # init a pretrained resnet
             num_target_classes = 10
-            self.feature_extractor = models.resnet50(
-                                        pretrained=True,
-                                        num_classes=num_target_classes)
+            self.feature_extractor = models.resnet50(pretrained=True)
             self.feature_extractor.eval()
 
             # use the pretrained model to classify cifar-10 (10 image classes)