Solved by el_samou_samou in post #6 
              
              
                copy past the source code from github and make some changes either in that part: 
self.classifier = nn.Sequential(
            nn.Linear(512 * 7 * 7, 4096),
            nn.ReLU(True),
            nn.Dropout(),
            nn.Linear(4096, 4096),
            nn.ReLU(True),
            nn.Dropout(),
  …