Update modelling_uniformer.py

modelling_uniformer.py

@@ -99,7 +99,6 @@ class CBlock(nn.Module):
 
         self.ls = layer_scale
         if self.ls:
-            print(f"Use layer_scale: {layer_scale}, init_values: {init_value}")
             self.gamma_1 = nn.Parameter(init_value * torch.ones((dim)),requires_grad=True)
             self.gamma_2 = nn.Parameter(init_value * torch.ones((dim)),requires_grad=True)
     
@@ -139,7 +138,6 @@ class SABlock(nn.Module):
         self.mlp = Mlp(in_features=dim, hidden_features=mlp_hidden_dim, act_layer=act_layer, drop=drop)
         self.ls = layer_scale
         if self.ls:
-            print(f"Use layer_scale: {layer_scale}, init_values: {init_value}")
             self.gamma_1 = nn.Parameter(init_value * torch.ones((dim)),requires_grad=True)
             self.gamma_2 = nn.Parameter(init_value * torch.ones((dim)),requires_grad=True)