From c5a29dbaec445aff1733917fe668bd038e3e4523 Mon Sep 17 00:00:00 2001 From: zh-hike <1583124882@qq.com> Date: Thu, 15 Dec 2022 11:56:11 +0000 Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E8=A7=84=E8=8C=83?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ppcls/arch/__init__.py | 2 +- ppcls/arch/backbone/model_zoo/wideresnet.py | 2 +- .../ssl/FixMatchCCSSL/FixMatchCCSSL_cifar10_4000_4gpu.yaml | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ppcls/arch/__init__.py b/ppcls/arch/__init__.py index cdf7196f..e8bdf53a 100644 --- a/ppcls/arch/__init__.py +++ b/ppcls/arch/__init__.py @@ -170,4 +170,4 @@ class AttentionModel(DistillationModel): else: out = self.model_list[idx](out, label) result_dict.update(out) - return result_dict + return result_dict \ No newline at end of file diff --git a/ppcls/arch/backbone/model_zoo/wideresnet.py b/ppcls/arch/backbone/model_zoo/wideresnet.py index cd9924dd..cbca228a 100644 --- a/ppcls/arch/backbone/model_zoo/wideresnet.py +++ b/ppcls/arch/backbone/model_zoo/wideresnet.py @@ -202,7 +202,7 @@ class Wide_ResNet(TheseusLayer): feat = self.relu(self.bn1(feat)) feat = F.adaptive_avg_pool2d(feat, 1) feat = paddle.reshape(feat, [-1, self.channels]) - + if self.proj: pfeat = self.fc1(feat) pfeat = self.relu_mlp(pfeat) diff --git a/ppcls/configs/ssl/FixMatchCCSSL/FixMatchCCSSL_cifar10_4000_4gpu.yaml b/ppcls/configs/ssl/FixMatchCCSSL/FixMatchCCSSL_cifar10_4000_4gpu.yaml index 90a17222..4b6381e8 100644 --- a/ppcls/configs/ssl/FixMatchCCSSL/FixMatchCCSSL_cifar10_4000_4gpu.yaml +++ b/ppcls/configs/ssl/FixMatchCCSSL/FixMatchCCSSL_cifar10_4000_4gpu.yaml @@ -103,7 +103,7 @@ DataLoader: order: hwc sampler: - name: DistributedBatchSampler # DistributedBatchSampler + name: DistributedBatchSampler batch_size: 64 drop_last: true shuffle: true -- GitLab