From cfe2ce6618ec69d4c1c25e2a7a3cb8994740d8c4 Mon Sep 17 00:00:00 2001 From: Xin Wen Date: Tue, 29 Oct 2019 15:50:49 +0800 Subject: [PATCH] changed fixed parameter "num_of_classes" fixed issues may occur when training PCB_RPP on a subset of Market-1501 --- reid/models/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/reid/models/resnet.py b/reid/models/resnet.py index aa9495d..b316854 100644 --- a/reid/models/resnet.py +++ b/reid/models/resnet.py @@ -40,7 +40,7 @@ def __init__(self, depth, pretrained=True, cut_at_pooling=False, mo.stride = (1,1) #================append conv for FCN==============================# self.num_features = num_features - self.num_classes = 751 #num_classes + self.num_classes = num_classes self.dropout = dropout out_planes = self.base.fc.in_features self.local_conv = nn.Conv2d(out_planes, self.num_features, kernel_size=1,padding=0,bias=False)