Update tiny_vit_sam.py

print
nickroh · Jun 30, 2023 · 6e54b7b · 6e54b7b
1 parent 7212d81
commit 6e54b7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mobile_sam/modeling/tiny_vit_sam.py b/mobile_sam/modeling/tiny_vit_sam.py
@@ -554,7 +554,7 @@ def set_layer_lr_decay(self, layer_lr_decay):
         # layers -> blocks (depth)
         depth = sum(self.depths)
         lr_scales = [decay_rate ** (depth - i - 1) for i in range(depth)]
-        #print("LR SCALES:", lr_scales)
+        print("LR SCALES:", lr_scales)
 
         def _set_lr_scale(m, scale):
             for p in m.parameters():