turn vgg from pretrained to random

2025-06-09 00:55:28 +08:00
parent f0b2e1b605
commit 9042815b34
1 changed files with 1 additions and 1 deletions
--- 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