Upload ResNetForZeroBitWatermarkDetection

modeling_resnet.py

@@ -17,7 +17,7 @@ class ResNetForZeroBitWatermarkDetection(ResNetPreTrainedModel):
             nn.ReLU(),
             nn.Dropout(0.2),
             nn.Linear(128, 1))
-        self.register_buffer('beta', torch.tensor([1.0]))
+        self.register_buffer('exp', torch.tensor([1.0]))
         
         # initialize weights and apply final processing
         self.post_init()
@@ -40,10 +40,10 @@ class ResNetForZeroBitWatermarkDetection(ResNetPreTrainedModel):
         x = self.classifier(pooled_output)
 
         # generalized-Gaussian recalibration, centering and scaling is already included in last linear layer
-        x =  0.5 + torch.sign(x) * 0.5 * torch.special.gammainc(1 / self.beta, torch.abs(x)**self.beta)
+        x =  0.5 + torch.sign(x) * 0.5 * torch.special.gammainc(1 / self.exp, torch.abs(x)**self.exp)
 
         # Laplacian calibration, centering and scaling is already included in last linear layer
-        # if beta==1
+        # if exp==1
         #x =  0.5 + torch.sign(x) * 0.5 * (1 - torch.exp(-torch.abs(x))) # laplacian
 
         logits = torch.log(x) - torch.log1p(-x)

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9e2334747e88820507d79d11ae3693a4c58292ba35b8d011e1a5585eba7fca9e
+oid sha256:2c29c45863f6bf8dc4f1db64a082cb7b9ce44ef163eb29a9ee476f8fe77a5b6f
 size 45051585