From 9042815b34024097c961fccd36bc8d36e96a4fdd Mon Sep 17 00:00:00 2001 From: Jiao77 Date: Mon, 9 Jun 2025 00:55:28 +0800 Subject: [PATCH] turn vgg from pretrained to random --- models/rord.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/rord.py b/models/rord.py index 17b94fd..a66fbfa 100644 --- a/models/rord.py +++ b/models/rord.py @@ -13,7 +13,7 @@ class RoRD(nn.Module): """ super(RoRD, self).__init__() - vgg16_features = models.vgg16(pretrained=True).features + vgg16_features = models.vgg16(pretrained=False).features # 共享骨干网络 self.slice1 = vgg16_features[:23] # 到 relu4_3